diff options
| author | junyanz <junyanz@berkeley.edu> | 2018-01-14 16:45:18 -0800 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2018-01-14 16:45:18 -0800 |
| commit | a956f8ca3521a23071d2d260bdcd59ccfb0772a1 (patch) | |
| tree | 97f08fc42a621d684d41d40f18925480966c7c07 /datasets/combine_A_and_B.py | |
| parent | fb533f82e4c8980701cb8860c5ba7f673f556791 (diff) | |
add 'aspect_ratio' in the test code
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)) |