valeoai / obsnet Goto Github PK

Hello ,

Thanks for your great work and your work has inspired me a lot.

I see the result is pretty good on Segment me if you can dataset, but when I follow your instruction and download the pre-trained model for Segment me if you can dataset, I got results like these: FPR@TPR95: 0.617; AUPRC: 0.569; AUROC: 0.862, which is much different than which is on the leaderboard: AUPRC: 0.754, FPR@TPR95: 0.267. (But I did the annotation of SMIYC myself, so the result may be different.) Is that possible that you can send me the anomaly score (.hdf5) for the Segment me if you can dataset, that you send to the organizer of SMIYC, so I can check if my result is right?

It'll be a lot of help for my current work!

Many Thanks

Hi, I have just cloned your code on the current version to conduct experiments on my own. I have download the pre-trained file from the provide google drive, and setup the dataset as instructed. However, when running the code with the following command (with some minor customization for the folder), I am facing several issues:

python main.py --dset_folder "data/CamVid" --segnet_file "pre_train/ObsNet_BddAnomaly.pth" --obsnet_file "./ckpt/camvid/" --data "CamVid" --tboard "./logs/camvid" --num_workers 4 --bsize 8 --adv "min_random_patch" --epsilon 0.025 --lr 0.2 --no_pretrained

1. Dataset.colors is not defined with the following error:

File "main.py", Line 32, In main
args.colors = np.array(train_loader.dataset.colors)
AttributeError: 'StreetHazard' object has no attribute 'colors'

I can easily tackle the problem by remove this line, but I wonder if there is something missing here.

2. Error in loading segnet_file:
   After temporally tackling the first issue, when running the executed code again, I find the error

File "~/miniconda3/envs/obsnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1672, in load_state_dict
self.__class__.__name__, "Init". join (error-msgs)))
RuntimeError: Error (s) in loading state_dict for SegNet:
Missing key(s) in state_dict: "down1.conv1.cbr_unit.O.weight", "down1.conv1.cbr_unit.O.bias",....
Unexpected key(s) in state _dict: "backbone.conv1.weight", "backbone.bn1.weight",....

Whatever the pre-trained model is (from the provide google drive), I still received the similar error. This erros is only happens when the pre-trained model is incompatible with the designed segnet network, leading to the load_state_dict error.

Therefore, I do not know if I am doing anything wrong here, and how to setup the right setting for running the code to train ObsNet either in CamVid or StreetHarzards dataset.

Thank you very much for your support