diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:36:48 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-04-18 12:36:48 +0200 |
| commit | 30f065f3a2d8d4dc11c457d271cc622fc30a812b (patch) | |
| tree | 4bfe7370480eba4032bc2952e22688aa47071cac /datasets/combine_A_and_B.py | |
| parent | 07fa49cff8f20bb8e3b7dd785464e79e21e0dd8e (diff) | |
| parent | d7ed19821513928670114ac18338ff3374ef49c4 (diff) | |
Merge branch 'master' of ghghgh.us:pix2pix
Diffstat (limited to 'datasets/combine_A_and_B.py')
| -rw-r--r-- | datasets/combine_A_and_B.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/datasets/combine_A_and_B.py b/datasets/combine_A_and_B.py index 70907ad..dd28f21 100644 --- a/datasets/combine_A_and_B.py +++ b/datasets/combine_A_and_B.py @@ -42,7 +42,9 @@ for sp in splits: if args.use_AB: name_AB = name_AB.replace('_A.', '.') # remove _A path_AB = os.path.join(img_fold_AB, name_AB) - im_A = cv2.imread(path_A, cv2.CV_LOAD_IMAGE_COLOR) - im_B = cv2.imread(path_B, cv2.CV_LOAD_IMAGE_COLOR) + #im_A = cv2.imread(path_A, cv2.CV_LOAD_IMAGE_COLOR) + #im_B = cv2.imread(path_B, cv2.CV_LOAD_IMAGE_COLOR) + im_A = cv2.imread(path_A, cv2.IMREAD_COLOR) + im_B = cv2.imread(path_B, cv2.IMREAD_COLOR) im_AB = np.concatenate([im_A, im_B], 1) cv2.imwrite(path_AB, im_AB) |