From 64fd519fb8bc45161eee7db3e7a447318349032a Mon Sep 17 00:00:00 2001 From: Dongyang Jin <73057174+jdyjjj@users.noreply.github.com> Date: Sun, 19 May 2024 21:51:14 +0800 Subject: [PATCH] Update pretreatment_rgb.py --- datasets/pretreatment_rgb.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/datasets/pretreatment_rgb.py b/datasets/pretreatment_rgb.py index 67e13b8..aec534a 100644 --- a/datasets/pretreatment_rgb.py +++ b/datasets/pretreatment_rgb.py @@ -44,11 +44,11 @@ def job(src, id): ratios.append(ratio) aligned_img = np.transpose(cv2.cvtColor(resize_with_padding(img, (256, 128)), cv2.COLOR_BGR2RGB), (2, 0, 1)) aligned_imgs.append(aligned_img) - if len(aligned_imgs) > 0: - output_path = os.path.join(DST, id, ty, vi) - os.makedirs(output_path, exist_ok=True) - pickle.dump(np.asarray(aligned_imgs), open(os.path.join(output_path, vi+"-aligned-rgbs.pkl"), "wb")) - pickle.dump(np.asarray(ratios), open(os.path.join(output_path, vi+"-ratios.pkl"), "wb")) + if len(aligned_imgs) > 0: + output_path = os.path.join(DST, id, ty, vi) + os.makedirs(output_path, exist_ok=True) + pickle.dump(np.asarray(aligned_imgs), open(os.path.join(output_path, vi+"-aligned-rgbs.pkl"), "wb")) + pickle.dump(np.asarray(ratios), open(os.path.join(output_path, vi+"-ratios.pkl"), "wb")) print('Successfully saved: ' + DST + '/' + id + '/' + ty + '/' + vi) if __name__ == '__main__':