when i run GLAS dataset,something has happended,how could i solve it? about kiu-net-pytorch HOT 5 CLOSED

Hi @wcoool , "Assertion t >= 0 && t < n_classes failed" - this happens because the ground truth segmentation maps contain other pixels than 0 or 1 ( in case of binary segmentation ). So, when you use GLAS dataset, before calculating loss, you can include a couple of lines of code where you make all the pixels above 127 in the ground truth as 1 and the rest as 0. This should solve the error.

y_batch[y_batch>=127] = 1
y_batch[y_batch<127] = 0

You can also do this alternatively by running a separate code to make all the ground truth segmentation labels 0 or 1 separately initially and using the new folder of labels as the train and test path.

from kiu-net-pytorch.

I am very glad that you could help me solve the problem. In the paper ,you said that you set 85 images for training and 80 images for testing in GLAS dataset. However, It has validation data in this code, Could you tell my that the numbers of training images, validation images and test images which you set in GLAS dataset.

from kiu-net-pytorch.

As the number of images is very less, only train and test datasets were used for the experiments. You can just give the test dataset directory in val_dataset variable.

from kiu-net-pytorch.

when i use GLAS dataset. the output of the prediction the array of the image is o ,how to solve it?

from kiu-net-pytorch.

Can you please check if you have processed the data according to the Notes ( meaning background is 0 and gland segmentation is 1) because I just retrained it and seems to work for me.

from kiu-net-pytorch.