the problem of function "predict_transform"

[jzh9830]

stride = inp_dim // prediction.size(2)
grid_size = inp_dim // stride
when inp_dim was not exactly divided by prediction.size(2), it may occur a error in prediction.view. Considering the grid_size equaling to the size of feature map, could we change it to "grid_size = prediction.size(2)" directly?

[samutamm]

The model seems to expect image of different size, which causes the problem. This #40 (comment) fixed it for me