Skip to content

Commit fc95a80

Browse files
committed
fx create_yaml file to adapt cvat
1 parent d59686e commit fc95a80

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

exports/yolov8/yolov8_export.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -104,5 +104,6 @@ def create_yaml(self, project):
104104
content = 'names:\n'
105105
for name in label_names:
106106
content += f'- {name}\n' # class mapping for helmet detection project
107-
content += f'nc: {len(label_names)}'
107+
content += f'path: ./\n'
108+
content += f'train: ./images\n'
108109
my_file.write(content)

0 commit comments

Comments
 (0)