diff --git a/main.py b/main.py index 2e87cc53259519f6d8152ee299312e649249b2f5..1f8aa2cd111ef786eedb797d91b8b85c690fe6b6 100644 --- a/main.py +++ b/main.py @@ -159,8 +159,9 @@ def net_eval(): batch_msk_lst.append(msk_) bi += 1 if bi == 4: - batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0], - base_crop_size=513, flip=True) + #batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0], + # base_crop_size=513, flip=True) + batch_res = batch_img_lst for mi in range(4): hist += cal_hist(batch_msk_lst[mi].flatten(), batch_res[mi].flatten(), 6) @@ -170,8 +171,9 @@ def net_eval(): print('processed {} images'.format(i + 1)) image_num = i if bi > 0: - batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0], - base_crop_size=513, flip=True) + # batch_res = eval_batch_scales(eval_net, batch_img_lst, scales=[1.0], + # base_crop_size=513, flip=True) + batch_res = batch_img_lst for mi in range(bi): hist += cal_hist(batch_msk_lst[mi].flatten(), batch_res[mi].flatten(), 6) print('processed {} images'.format(image_num + 1))