luxonis / datadreamer

Creation of annotated datasets from scratch using Generative AI and Foundation Computer Vision models
Apache License 2.0
77 stars 5 forks source link

Batch annotation #35

Closed HonzaCuhel closed 8 months ago

HonzaCuhel commented 8 months ago

This PR includes adding batched annotation.

Some benchmarks (NVIDIA L4 24GB GPU):

Set the default batch size to 1.

github-actions[bot] commented 8 months ago

Test Results

  6 files    6 suites   45m 17s :stopwatch:  72 tests  30 :white_check_mark:  42 :zzz: 0 :x: 432 runs  180 :white_check_mark: 252 :zzz: 0 :x:

Results for commit a9eb3445.

:recycle: This comment has been updated with latest results.

github-actions[bot] commented 8 months ago

☂️ Python Coverage

current status: ✅

Overall Coverage

Lines Covered Coverage Threshold Status
820 391 48% 0% 🟢

New Files

No new covered files...

Modified Files

File Coverage Status
datadreamer/dataset_annotation/image_annotator.py 92% 🟢
datadreamer/dataset_annotation/owlv2_annotator.py 72% 🟢
datadreamer/dataset_annotation/utils.py 40% 🟢
datadreamer/pipelines/generate_dataset_from_scratch.py 44% 🟢
TOTAL 62% 🟢

updated for commit: a9eb344 by action🐍

sokovninn commented 8 months ago

@HonzaCuhel could you please add missing docstrings and replace annotate() with annotate_batch() in the examples?

HonzaCuhel commented 8 months ago

Sure thing, I have forgot about it

HonzaCuhel commented 8 months ago

@sokovninn I've updated the example and added the missing docstrings.