Problems with infer

[Codermay]

Thanks for your good implementation. I have run your code on my own dataset. And I want to know, do your code provide multiple images inferring?

[potterhsu]

No, but you can easily extend for your needs.

[PaperLamp]

Thanks for your codes. May I ask for your help about the following problem:
When I try to evaluate or infer, I meet this error:
1.evaluate:
size mismatch for detection._proposal_transformer.bias: copying a param with shape torch.Size([368]) from checkpoint, the shape in current model is torch.Size([84]).
2.infer:
size mismatch for rpn._anchor_transformer.weight: copying a param with shape torch.Size([48, 512, 1, 1]) from checkpoint, the shape in current model is torch.Size([36, 512, 1, 1]).
Thank you

[PeterBishop0]

same problems,have you worked it out?@PaperLamp

[PeterBishop0]

@potterhsu

[potterhsu]

These sorts of errors usually occurred when your configs between training and inference are inconsistent, for example:

• You were training with anchor_sizes=[64,128,256,512]
• But are inferring with anchor_sizes=[128,256,512]