From cd2ac9c254c8fefeeb2bb4fdf065ea2ee42a3291 Mon Sep 17 00:00:00 2001 From: GoldenAnpu Date: Mon, 23 Dec 2024 01:12:39 +0100 Subject: [PATCH] Fix image_ids list for only annotations selected_output --- src/main.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/main.py b/src/main.py index 9fce0e9..545492c 100644 --- a/src/main.py +++ b/src/main.py @@ -56,11 +56,10 @@ def process(self, context: sly.app.Export.Context): images = [ image for image in images if image.labels_count > 0 or len(image.tags) > 0 ] - + image_ids = [image_info.id for image_info in images] coco_ann = {} if selected_output == "images": - image_ids = [image_info.id for image_info in images] paths = [os.path.join(img_dir, image_info.name) for image_info in images] di_pbar = sly.tqdm_sly(desc=f"Downloading images", total=len(image_ids))