search

davidtvs / PyTorch-ENet

PyTorch implementation of ENet
MIT License
383 stars 129 forks source link

something wrong with mIoU #37

Closed isalirezag closed 4 years ago

isalirezag commented 4 years ago

For some reason, I received a negative number for mIoU. I am not sure what is wrong with it

image

davidtvs commented 4 years ago

hmm weird...what dataset are you running and with what command-line options?

isalirezag commented 4 years ago

Im using cityscapes, have not change anything, so most of the stuff are the default ones. I am also looking into it to see if I can figure it out.

davidtvs commented 4 years ago

I would suggest looking into what's happening in the IoU class during training.

As for command-line options, you should have something like this:

--mode train
--batch_size 4
--dataset cityscapes
--dataset_dir <path_to_cityscapes_root>
isalirezag commented 4 years ago

Thanks! yes, the command line is similar to what provided. Yup, I am looking into it, I will let you know when I figure that out.

isalirezag commented 4 years ago

ran the code several times and did not observe that behavior anymore, so I am going to close this issue.