Update get_train_val_YOLO.py

get_train_val_YOLO.py

@@ -110,4 +110,4 @@ with open(str(direction+'custom_data.yaml'),'w') as f:
 	f.write('train : '+direction+'images/train\n')
 	f.write('test : '+direction+'images/val\n')
 	f.write('nc : ' + str(len(tab))+'\n')
-	f.write('names : ' + str(tab.columns))
+	f.write('names : ' + str(tab.index.tolist()))