Merge pull request #229 from roboflow/yolov9

YOLOv9 Export Format

Author: tonylampada

CLI-COMMANDS.md

@@ -57,7 +57,7 @@ positional arguments:
 options:
   -h, --help   show this help message and exit
   -f FORMAT    Specify the format to download the version. Available options: [coco, yolov5pytorch, yolov7pytorch, my-yolov6, darknet,
-               voc, tfrecord, createml, clip, multiclass, coco-segmentation, yolo5-obb, png-mask-semantic, yolov8]
+               voc, tfrecord, createml, clip, multiclass, coco-segmentation, yolo5-obb, png-mask-semantic, yolov8, yolov9]
   -l LOCATION  Location to download the dataset
 ```
 

roboflow/__init__.py

@@ -14,7 +14,7 @@
 from roboflow.models import CLIPModel, GazeModel  # noqa: F401
 from roboflow.util.general import write_line
 
-__version__ = "1.1.19"
+__version__ = "1.1.20"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/version.py

@@ -714,7 +714,7 @@ def data_yaml_callback(content: dict) -> dict:
                 content["train"] = location + content["train"].lstrip(".")
                 content["val"] = location + content["val"].lstrip(".")
                 content["test"] = location + content["test"].lstrip(".")
-            if format in ["yolov5pytorch", "yolov7pytorch", "yolov8"]:
+            if format in ["yolov5pytorch", "yolov7pytorch", "yolov8", "yolov9"]:
                 content["train"] = location + content["train"].lstrip("..")
                 content["val"] = location + content["val"].lstrip("..")
             try:
@@ -729,7 +729,7 @@ def data_yaml_callback(content: dict) -> dict:
                 pass
             return content
 
-        if format in ["yolov5pytorch", "mt-yolov6", "yolov7pytorch", "yolov8"]:
+        if format in ["yolov5pytorch", "mt-yolov6", "yolov7pytorch", "yolov8", "yolov9"]:
             amend_data_yaml(path=data_path, callback=data_yaml_callback)
 
     def __str__(self):

roboflow/roboflowpy.py

@@ -164,7 +164,7 @@ def _add_download_parser(subparsers):
         help="Specify the format to download the version. Available options: [coco, "
         "yolov5pytorch, yolov7pytorch, my-yolov6, darknet, voc, tfrecord, "
         "createml, clip, multiclass, coco-segmentation, yolo5-obb, "
-        "png-mask-semantic, yolov8]",
+        "png-mask-semantic, yolov8, yolov9]",
     )
     download_parser.add_argument("-l", dest="location", help="Location to download the dataset")
     download_parser.set_defaults(func=download)