ScanSSD
ScanSSD copied to clipboard
RuntimeError: shape '[8, 38227, 2]' is invalid for input of size 734832
I'm getting this error when following the steps outlined in the README.md file for testing pre-trained model weights on testing_data.
Did you find a solution for this? Facing the same error.
self.num_classes).transpose(2, 1) RuntimeError: shape '[2, 38227, 2]' is invalid for input of size 183708
Did you find a solution for this? Facing the same error.
The padding in the LOC and CONF layers of multibox architecture causes the above error. Specifying padding to (0,2) solved the issue for me.