From a956f8ca3521a23071d2d260bdcd59ccfb0772a1 Mon Sep 17 00:00:00 2001 From: junyanz Date: Sun, 14 Jan 2018 16:45:18 -0800 Subject: add 'aspect_ratio' in the test code --- datasets/combine_A_and_B.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) (limited to 'datasets') diff --git a/datasets/combine_A_and_B.py b/datasets/combine_A_and_B.py index 4d1e2a2..70907ad 100644 --- a/datasets/combine_A_and_B.py +++ b/datasets/combine_A_and_B.py @@ -1,4 +1,3 @@ -from pdb import set_trace as st import os import numpy as np import cv2 @@ -21,7 +20,7 @@ for sp in splits: img_fold_A = os.path.join(args.fold_A, sp) img_fold_B = os.path.join(args.fold_B, sp) img_list = os.listdir(img_fold_A) - if args.use_AB: + if args.use_AB: img_list = [img_path for img_path in img_list if '_A.' in img_path] num_imgs = min(args.num_imgs, len(img_list)) -- cgit v1.2.3-70-g09d2