Skip to content

Commit 9f1c94c

Browse files
committed
Final exporter fix
1 parent c7cc4cb commit 9f1c94c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

agml/data/exporters/yolo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def export_yolo(dataset, yolo_path=None):
121121
f.write("\n".join(split_images))
122122

123123
# update the YAML file with the corresponding locs of train/val/test
124-
yaml_dump[split_name] = os.path.join(yolo_data_path, f'{split_name}.txt')
124+
yaml_dump[split_name] = f'{split_name}.txt'
125125

126126
for loader_name, loader_content in loader_contents.items():
127127
for image, annotation_set in loader_content.items():

0 commit comments

Comments
 (0)