problem in convert_to_onehot in model_vnet3d_multilabel.py

[anirban1513]

Traceback (most recent call last):
File "train.py", line 35, in
trainVnet()
File "train.py", line 31, in trainVnet
[16, 8])
File "/home/ubuntu/new_lobe/model_vnet3d_multilabel.py", line 322, in train
batch_ys_onehot = convert_to_one_hot(batch_ys, self.numclass)
File "/home/ubuntu/new_lobe/model_vnet3d_multilabel.py", line 190, in convert_to_one_hot
one_hoty = np.eye(numclass)[one_hoty.reshape(-1).astype(np.int)]
IndexError: index 8 is out of bounds for axis 0 with size 7

[junqiangchen]

the number class is 7,but your is 8,there is different.

[Jishna04]

I am also having the same issue. I have attached the screenshot. can you please help me with that.[](url

)