summaryrefslogtreecommitdiff
path: root/datasets/combine_A_and_B.py
diff options
context:
space:
mode:
authorjules@lens <julescarbon@gmail.com>2018-04-18 12:35:19 +0200
committerjules@lens <julescarbon@gmail.com>2018-04-18 12:35:19 +0200
commitd7ed19821513928670114ac18338ff3374ef49c4 (patch)
tree05394bf708ad03283834e8fcd3a6ea802e82aad2 /datasets/combine_A_and_B.py
parent1fd836eb35924e9771253ad6fa8f74ed16a62f0b (diff)
fix opencv bug
Diffstat (limited to 'datasets/combine_A_and_B.py')
-rw-r--r--datasets/combine_A_and_B.py6
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)