Add images without annotations to LuxonisDataset

.github/workflows/gar-publish-dev.yaml

@@ -15,9 +15,7 @@ jobs:
 
     steps:
       - name: 'Checkout GitHub Action'
-        uses: actions/checkout@main
-        with:
-          ref: dev  # Checkout the dev branch
+        uses: actions/checkout@v4
 
       - id: 'auth'
         name: 'Authenticate to Google Cloud'
@@ -36,5 +34,5 @@ jobs:
       - name: 'Build Inventory Image'
         working-directory: .
         run: |
-          docker build --build-arg GITHUB_TOKEN=${{secrets.GHCR_PAT}} --build-arg BRANCH=dev . --tag $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer:dev
+          docker build --build-arg GITHUB_TOKEN=${{secrets.GHCR_PAT}} --build-arg BRANCH=${{ github.ref_name }} . --tag $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer:dev
           docker push $GAR_LOCATION-docker.pkg.dev/$PROJECT_ID/internal/datadreamer --all-tags

datadreamer/utils/luxonis_dataset_converter.py

@@ -80,6 +80,12 @@ def dataset_generator():
                 image_full_path = os.path.join(dataset_dir, image_path)
                 width, height = Image.open(image_full_path).size
                 labels = data[image_path]["labels"]
+
+                if len(labels) == 0:
+                    yield {
+                        "file": image_full_path,
+                    }
+
                 for label in labels:
                     yield {
                         "file": image_full_path,

requirements.txt

@@ -12,6 +12,6 @@ accelerate>=0.25.0
 scipy>=1.10.0
 bitsandbytes>=0.42.0
 nltk>=3.8.1
-luxonis-ml[all]>=0.4.1
+luxonis-ml[all]>=0.5.0
 python-box>=7.1.1
 gcsfs>=2023.1.0