search

Deci-AI / data-gradients

Computer Vision dataset analysis
Apache License 2.0
293 stars 33 forks source link

Hotfix/dg 000 fix empty batch detection #160

Closed Louis-Dupont closed 1 year ago

Louis-Dupont commented 1 year ago

Small fix in DG which can break the detection pipeline under the combination of 3 factors:

  1. Working with batches
  2. One batch does not include any target
  3. The empty label does not respect the expected format (e.g. tensor([], size=(BS, 0)) instead of tensor([], size=(BS, 0, 5)))