diff --git a/main.py b/main.py index 1a20473ed303b10390eca4e4d26dbab64bb691dc..4339cf174ba70d1e433e8604d9638066b22fee6a 100644 --- a/main.py +++ b/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]