Skip to content

Commit

Permalink
Fix image_ids list for only annotations selected_output
Browse files Browse the repository at this point in the history
  • Loading branch information
GoldenAnpu committed Dec 23, 2024
1 parent c5a4f78 commit cd2ac9c
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit cd2ac9c

Please sign in to comment.