From ae6ec401354ac525a49cd399c8b42172988fa7ec Mon Sep 17 00:00:00 2001 From: Maxim Zhiltsov Date: Wed, 2 Sep 2020 22:07:08 +0300 Subject: [PATCH] Image copying in CVAT format (#2091) --- cvat/apps/dataset_manager/formats/cvat.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/cvat/apps/dataset_manager/formats/cvat.py b/cvat/apps/dataset_manager/formats/cvat.py index ae1257c2..962138c7 100644 --- a/cvat/apps/dataset_manager/formats/cvat.py +++ b/cvat/apps/dataset_manager/formats/cvat.py @@ -530,15 +530,16 @@ def _export(dst_file, task_data, anno_callback, save_images=False): frame_provider = FrameProvider(task_data.db_task.data) frames = frame_provider.get_frames( frame_provider.Quality.ORIGINAL, - frame_provider.Type.NUMPY_ARRAY) + frame_provider.Type.BUFFER) for frame_id, (frame_data, _) in enumerate(frames): frame_name = task_data.frame_info[frame_id]['path'] - if '.' in frame_name: - save_image(osp.join(img_dir, frame_name), - frame_data, jpeg_quality=100, create_dir=True) - else: - save_image(osp.join(img_dir, frame_name + '.png'), - frame_data, create_dir=True) + ext = '' + if not '.' in osp.basename(frame_name): + ext = '.png' + img_path = osp.join(img_dir, frame_name + ext) + os.makedirs(osp.dirname(img_path), exist_ok=True) + with open(img_path, 'wb') as f: + f.write(frame_data.getvalue()) make_zip_archive(temp_dir, dst_file)