easy-faster-rcnn.pytorch
easy-faster-rcnn.pytorch copied to clipboard
Problems with infer
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?
No, but you can easily extend for your needs.
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
same problems,have you worked it out?@PaperLamp
@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]