diff options
Diffstat (limited to 'datasets/combine_A_and_B.py')
| -rw-r--r-- | datasets/combine_A_and_B.py | 3 |
1 files changed, 1 insertions, 2 deletions
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)) |