working on google colab about anomaly-seg HOT 6 CLOSED

train.py needs to be modified as per this issue to run on a single GPU.

to run on a single GPU it's

train.py --gpu 0 

as opposed to train.py --gpu 1 .

I've also uploaded the pretrained models that can be found here

Finally to run the eval_ood.py if using the default config/ade20k-resnet50dilated-ppm_deepsup.yaml and the model I've uploaded, change the num_classes parameter in the yaml file to be 13.

Thanks for pointing out the typo. I fixed it and another as well.

I've also updated the README as well to reflect these changes

from anomaly-seg.

from anomaly-seg.

Note that I follow the recommanded steps in the README and actually it may come from python create_dataset.py step, I have 'FileNotFoundError: [Errno 2] No such file or directory: 'data/train/images/test/' ... Could you give some more indications on how the repositories should be ?

from anomaly-seg.

@xksteven

from anomaly-seg.

I updated the repo. The error should be gone now. Pull it again or download the updated create_dataset.py and defaults file. Thanks for the bug report.

@julienguegan

from anomaly-seg.

Thanks for the update @xksteven . The previous error have disappeared but I think there is still something missing because I can't make the command python train.py works... I had to add feed_dict = feed_dict[0] in line 32 of models.py because I had a list and not a dictionnary but then I have "Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same". You can check link_colab if you have time

PS : I also noted one minor error in eval_ood.py : in_scores = - conf[np.logical_not(out_label))] line 51, there is one parenthesis too many

from anomaly-seg.