SSDSpec for image size smaller than 300

qfgaohao / pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

https://medium.com/@smallfishbigsea/understand-ssd-and-implement-your-own-caa3232cd6ad

MIT License

1.39k stars 529 forks source link

SSDSpec for image size smaller than 300 #171

Open AmitAilianiSDC opened 2 years ago

AmitAilianiSDC commented 2 years ago

I am running into an error if I change the image size from 300x300 to something smaller or bigger.

How can we set ssd spec for custom image size ?

IndexError: The shape of the mask at index 1 does not match the shape of the indexed tensor at index 1

likecaulfield commented 8 months ago

Did you solve this problem? I'm having exactly the same issue ...