Traceback (most recent call last):
File "test.py", line 137, in
test_net(net, imdb, max_per_image, thresh, vis)
File "test.py", line 71, in test_net
size_index
File "/home/hyunbae/yolo2-pytorch/utils/yolo.py", line 119, in postprocess
assert bbox_pred.shape[0] == 1, 'postprocess only support one image per batch' # noqa
AssertionError: postprocess only support one image per batch
it says I have to prove one image per batch but I am not sure how to handle this
Traceback (most recent call last): File "test.py", line 137, in
test_net(net, imdb, max_per_image, thresh, vis)
File "test.py", line 71, in test_net
size_index
File "/home/hyunbae/yolo2-pytorch/utils/yolo.py", line 119, in postprocess
assert bbox_pred.shape[0] == 1, 'postprocess only support one image per batch' # noqa
AssertionError: postprocess only support one image per batch
it says I have to prove one image per batch but I am not sure how to handle this