INFINIQ-AI1 / RISF

official implementation of "Re-scoring using image-language similarity for few-shot object detection" CVIU, April 2024
Other
9 stars 0 forks source link

fine tuning ERROR #1

Open swc0717 opened 7 months ago

swc0717 commented 7 months ago

Do I need to change OUT_DIR and WEIGHTS in the configuration file when running the fine-tuning commands? If you do not change it, you will not find the file, and after changing it, you will report an error, and the error content is KeyError: 4

INFINIQ-AI1 commented 7 months ago

Yes, you need to change OUT_DIR to save your model to any directory, and WEIGHTS to your pretrained weights directory. If your device does not have pretrained weights, you can simply use WEIGHTS: "Detectron2://ImageNetPretrained/MSRA/R-101.pkl". Please upload the full code log for KeyError:4.

swc0717 commented 6 months ago

Thank you very much for your reply. This is the detailed error information about the key-value pairs of the dictionary 1 ![Uploading 1.jpg…]()

INFINIQ-AI1 commented 6 months ago

Hi @swc0717 ! 😄 It looks like 2 is not included in self.class_mapper. First, you can check whether "coco14_test_novel" is the test dataset in the configuration file. If you use a custom dataset, you need to modify the class_mapper in "risf/evaluation/cmclip.py". Second, please print the build process of class_mapper (42~45 of risf/evaluation/cmclip.py). You will find out what the problem is.

qjh666888 commented 3 weeks ago

Excuse me, has it been resolved