RLIPv2
RLIPv2 copied to clipboard
Can't figure out how to run inference with pre-trained weights
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?