qfgaohao/pytorch-ssd

SSDSpec for image size smaller than 300

Opened this issue · 1 comments

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

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