查看img_的shape

main.py

@@ -102,6 +102,7 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True):
         # resize_h, resize_w, _ = img_.shape
         # hwc to chw
         # img_ = img_.transpose((2, 0, 1))
+        print(img_.shape)
         img_, resize_h, resize_w = pre_process(img_, crop_size)
         batch_img[l] = img_
         resize_hw.append([resize_h, resize_w])