diff --git a/main.py b/main.py index 3ff9183dfebeb6de8f07c5ab2ddc1bb6d66e527d..db06a5e2c2682389cec3c59f4cffbce446b46530 100644 --- a/main.py +++ b/main.py @@ -117,11 +117,12 @@ def eval_batch(eval_net, img_lst, crop_size=513, flip=True): def eval_batch_scales(eval_net, img_lst, scales, base_crop_size=513, flip=True): + print(scales) sizes_ = [int((base_crop_size - 1) * sc) + 1 for sc in scales] - probs_lst = eval_batch(args, eval_net, img_lst, crop_size=sizes_[0], flip=flip) + probs_lst = eval_batch(eval_net, img_lst, crop_size=sizes_[0], flip=flip) print(sizes_) for crop_size_ in sizes_[1:]: - probs_lst_tmp = eval_batch(args, eval_net, img_lst, crop_size=crop_size_, flip=flip) + 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] @@ -145,7 +146,6 @@ def net_eval(): bi = 0 image_num = 0 for i, line in enumerate(img_lst): - print(line) img_path = os.path.join('./datasets/JPEGImages/', line+'.bmp') msk_path = os.path.join('./datasets/SegmentationClass/', line+'.png') img_ = cv2.imread(img_path)