ModuleNotFoundError: No module named 'pretrainedmodels'

[arbezlo]

Dear authors,

While wanting to experiment with your repo I found myself in a bit of a deadend. I wanted to reproduce an evaluation of the llvip dataset as suggested in the README. After dowloading the weights you provided. I executed the following command :

CUDA_VISIBLE_DEVICES=0 python eval_hallucidet.py --pretrained --modality ir --detector-path /scratch2/clear/larbez/Workspace/Hallucidet/checkpoints/llvip/seed123/fasterrcnn_rgb_llvip_seed123.ckpt --hallucidet-path /scratch2/clear/larbez/Workspace/Hallucidet/checkpoints/llvip/seed123/hallucidet_llvip_seed123.ckpt --wandb-project wacv2024 --wandb-name detector_fasterrcnn_hallucidet_det01reg01_llvip_200ep_seed123 --detector fasterrcnn --dataset llvip --epochs 1 --batch 8 --seed 123

Which resulted in the following error :

After checking the whole repository it seems that the pretrainedmodels folder is missing. I tried working around it but I didn't manage to get rid of that error. Could you provide me with a fix of some sort ?

Thanks in advance for your help.

[heitorrapela]

Hello @arbezlo, thanks for your interesting in our work!

Could you try to install the pretrainedmodels?

pip install pretrainedmodels==0.7.4

Let me know if this worked for you, best regards.