search

heitorrapela / HalluciDet

[WACV2024] HalluciDet: Hallucinating RGB Modality for Person Detection Through Privileged Information (Accepted at WACV 2024 and LatinX@CVPR2024 Extended Abstract)
MIT License
18 stars 0 forks source link

ModuleNotFoundError: No module named 'pretrainedmodels' #4

Open arbezlo opened 2 weeks ago

arbezlo commented 2 weeks ago

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 :

Capture d’écran de 2024-11-04 11-33-46

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 commented 2 weeks ago

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.