search

ZhanYang-nwpu / RSVG-pytorch

RSVG: Exploring Data and Model for Visual Grounding on Remote Sensing Data, 2022
106 stars 4 forks source link

Weight file error ? #14

Closed Hasanmog closed 1 year ago

Hasanmog commented 1 year ago

Firstly, good job! I was testing your model on the RSVGD dataset, but got the following problem when loading the weight file.

!python RSVG-pytorch/main.py --images_path "/content/drive/MyDrive/JPEGImages" --anno_path "/content/drive/MyDrive/Annotations" --pretrain "/content/drive/MyDrive/detr-r50-e632da11.pth" --test

`pretrained_dict = checkpoint['state_dict']

KeyError: 'state_dict'`

According to what I searched about the cause of this error , it is either the weight file is corrupted or missing parameters. Any Idea ?

aghand0ur commented 1 year ago

The authors only provided the pre-trained weights. However, the resultant trained model checkpoints are not available.

Any plan to release them soon @ZhanYang-nwpu?

aghand0ur commented 1 year ago

The above error can be avoided using the following as noted by @LANMNG: python RSVG-pytorch/main.py --images_path "/content/drive/MyDrive/dataset/RSVG_dataset/Images/JPEGImages" --anno_path "/content/drive/MyDrive/dataset/RSVG_dataset/Annotations/Annotations" --pretrain "" --test