Merge pull request #580 from DagsHub/bug/yolo-export

Bug: Don't add "images" to the annotations for yolo export
DagsHub · Jan 9, 2025 · d37cd0e · d37cd0e
2 parents 9e20311 + 68ff74e
commit d37cd0e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/dagshub/data_engine/model/query_result.py b/dagshub/data_engine/model/query_result.py
@@ -752,8 +752,6 @@ def export_as_yolo(
 
         # If there's no folder "images" in the datasource, prepend it to the path
         if not any("images/" in ann.filename for ann in annotations):
-            for ann in annotations:
-                ann.filename = os.path.join("images", ann.filename)
             image_download_path = Path(download_dir) / "images"
 
         context = YoloContext(annotation_type=annotation_type, categories=categories)

diff --git a/setup.py b/setup.py
@@ -41,7 +41,7 @@ def get_version(rel_path: str) -> str:
     "python-dateutil",
     "boto3",
     "semver",
-    "dagshub-annotation-converter>=0.1.0",
+    "dagshub-annotation-converter>=0.1.3",
 ]
 
 extras_require = {