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 https://github.com/ayooshkathuria/YOLO_v3_tutorial_from_scratch/issues/40#issuecomment-516392022 fixed it for me