RLIPv2 icon indicating copy to clipboard operation
RLIPv2 copied to clipboard

Can't figure out how to run inference with pre-trained weights

Open lunaluxie opened this issue 6 months ago • 2 comments

Hello,

Excellent work! I'm trying to adapt it to another dataset, but I am struggling to find out how to load pretrained weights.

Specifically, I'm looking at the fully fined-tuned weights on HICO-DET for the model RLIPv2-ParSeDA with backbone ResNet-50.

I have downloaded the weights, and thought I could run inference with the command

python3 inference_on_custom_imgs_hico.py --batch_size 1 --param_path RLIP_PDA_v2_HICO_R50_VGCOO365_COO365det_RQL_LSE_RPL_20e_L1_20e_checkpoint0019.pth --save_path out --backbone resnet50 --RLIP_ParSeDA_v2

But I get an error saying that a bunch of keys are missing in the state_dict. I have tried to look at the different arguments, but I can't figure out what the appropriate command is.

Do you have any documentation that specifies how to load the appropriate model for the different pretrained weights?

lunaluxie avatar Aug 22 '24 18:08 lunaluxie