Closed kyoto7250 closed 2 years ago
You can download the pretrained detectron2 model via this link.
For more models, you can check the model zoo of detectron2.
@Poeroz
I seem this model can not work.
I changed the name of the model you linked to to "model_weights_path, 'faster_rcnn_R_101_C4_3x.pkl".
# detectron2/model_weights
wget https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_101_C4_3x/138204752/model_final_298dad.pkl
# rename
mv model_final_298dad.pkl faster_rcnn_R_101_C4_3x.pkl
Then, I run extract_features.py
and this error happened.
cd visual_grounding/
python3 extract_features.py
Is these the correct the model? It would be helpful if you could also tell us your version of detectron2🙏🏻
my environment
Thank you.
And config file is here
P.S It seems that config and load specify different models.
@Poeroz
friendly mention :)
Would you be willing to share your opinion? I can't reproduce it as it is...
Change 'extract_features.py' file line 49 to '_default="detectron2/configs/COCO-InstanceSegmentation/mask_rcnn_R_101_C43x.yaml",'
Thanks for the quick response the other day!
I am now trying to run the visual_grounding directory. I found pretrained bert model in google drive, and yolo v3 model in the script, but I missed pretrained detectron2 model.
I think it's needed to extract features from the bbox. Where is the pretrained detectron2 model?
I'm sorry for asking so many questions 🙏🏻
Best regards.