Open Uspectacle opened 3 years ago
Indeed models trained with deformable Transformers currently do not support the generation of attention maps. I added an assert statement and opened an issue https://github.com/fundamentalvision/Deformable-DETR/issues/80 as this is not even supported by the official deformable DETR repository.
Furthermore, I uploaded new model files and fixed the code. Now you should be able to generate attention maps with the provided segmentation model. This is only possible since this model works based on vanilla DETR.
Where can I download the pretrained model?thanks~
Step 4 here.
Hello,
I try to get attention maps from pre-trained models updated 1 hours ago. Using a virtual environment with all the requirements from INSTALL.md in the same version, I tryed executing the following :
I got :
With the same but with obj_detect_checkpoint_file=models/mots20_train_masks/checkpoint.pth instead I got :
By adding focal_loss=false and focal_alpha=0.25 to models/mots20_train_masks/config.yaml, I now get :
Thanks you for this excellent paper, for releasing the code and for your time.