查看img_的shape

main.py

@@ -130,6 +130,7 @@ def eval_batch_scales(eval_net, img_lst, scales,
     sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales]
     probs_lst = eval_batch(eval_net, img_lst, crop_size=sizes_[0], flip=flip)
     for crop_size_ in sizes_[1:]:
+        print(img_lst.shape)
         probs_lst_tmp = eval_batch(eval_net, img_lst, crop_size=crop_size_, flip=flip)
         for pl, _ in enumerate(probs_lst):
             probs_lst[pl] += probs_lst_tmp[pl]