Pytorch-UNet icon indicating copy to clipboard operation
Pytorch-UNet copied to clipboard
verified publisher icon milesial

The output image is strange : train.py or predict.py problem?

Open t19b205d opened this issue 3 years ago • 3 comments

Hello! My English isn't so good so feel free to ask me if there is anything unclear. I trained 2classes. using dataset is CT image of lung tumor.

The mask image below is wandb/run-20221111_*************/files/media/images/pred_***********.png background:0 ,tumor:1 are classified pred_22221_6b9f942956941162fa97

But output mask images is strange. output05003

I want to output like wandb/run-20221111_*************/files/media/images.

t19b205d avatar Nov 30 '22 06:11 t19b205d

Hi, can you try the latest master version?

milesial avatar Dec 06 '22 19:12 milesial

It still results the same

mahendran-narayanan avatar Dec 07 '22 09:12 mahendran-narayanan

Can you share the pretrained model along with the input image that results in these masks?

milesial avatar Dec 07 '22 11:12 milesial