Skip to content

Commit 7dc43e0

Browse files
committed
fix resize.
1 parent 17e2a2d commit 7dc43e0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

resize_images.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -36,15 +36,15 @@ def resize_data_ade(record, verbose):
3636
h_new, w_new = round(h / float(w) * max_size), max_size
3737
else:
3838
return 0
39-
cv2.imwrite(img_path, cv2.resize(img, (h_new, w_new),
39+
cv2.imwrite(img_path, cv2.resize(img, (w_new, h_new),
4040
interpolation=cv2.INTER_LINEAR))
4141
seg_obj_path = md["seg_filename"]
4242
seg = cv2.imread(seg_obj_path)
43-
cv2.imwrite(seg_obj_path, cv2.resize(seg, (h_new, w_new),
43+
cv2.imwrite(seg_obj_path, cv2.resize(seg, (w_new, h_new),
4444
interpolation=cv2.INTER_NEAREST))
4545
for i, seg_part_path in enumerate(md["part_filenames"]):
4646
seg = cv2.imread(seg_part_path)
47-
cv2.imwrite(seg_part_path, cv2.resize(seg, (h_new, w_new),
47+
cv2.imwrite(seg_part_path, cv2.resize(seg, (w_new, h_new),
4848
interpolation=cv2.INTER_NEAREST))
4949
return 0
5050

0 commit comments

Comments
 (0)