Skip to content

Commit 9415998

Browse files
yeldarbyBrad Dwyer
andauthored
Update YAML for YOLOv8 (#94)
Co-authored-by: Brad Dwyer <[email protected]>
1 parent 95d7085 commit 9415998

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

roboflow/core/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -488,7 +488,7 @@ def __reformat_yaml(self, location, format):
488488
489489
:return None:
490490
"""
491-
if format in ["yolov5pytorch", "yolov7pytorch"]:
491+
if format in ["yolov5pytorch", "yolov7pytorch", "yolov8"]:
492492
with open(location + "/data.yaml") as file:
493493
new_yaml = yaml.safe_load(file)
494494
new_yaml["train"] = location + new_yaml["train"].lstrip("..")

0 commit comments

Comments
 (0)