Skip to content
Snippets Groups Projects
Commit f3d0a488 authored by Stephane Chavin's avatar Stephane Chavin
Browse files

Update get_train_val_YOLO.py

parent adbd14fc
No related branches found
No related tags found
No related merge requests found
...@@ -118,6 +118,6 @@ liste_espece = pd.read_csv(str(path+'../liste_especes.csv')) ...@@ -118,6 +118,6 @@ liste_espece = pd.read_csv(str(path+'../liste_especes.csv'))
with open(str(direction+'custom_data.yaml'),'w') as f: with open(str(direction+'custom_data.yaml'),'w') as f:
f.write('train : '+direction+'images/train\n') f.write('train : '+direction+'images/train\n')
f.write('test : '+direction+'images/val\n') f.write('val : '+direction+'images/val\n')
f.write('nc : ' + str(len(liste_espece))+'\n') f.write('nc : ' + str(len(liste_espece))+'\n')
f.write('names : ' + liste_espece.espece.tolist()) f.write('names : ' + liste_espece.espece.tolist())
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment