problem in convert_to_onehot in model_vnet3d_multilabel.py

junqiangchen / LungLobeSegmentation

Lobe Segmentation

https://cirrus.grand-challenge.org/LobeSegmentation/

30 stars 14 forks source link

problem in convert_to_onehot in model_vnet3d_multilabel.py #5

Open anirban1513 opened 3 years ago

anirban1513 commented 3 years ago

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 commented 3 years ago

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

Jishna04 commented 2 years ago

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