Update get_train_val_YOLO.py

get_train_val_YOLO.py

@@ -58,7 +58,7 @@ for i in tqdm(os.listdir(path)):
 df.reset_index(inplace = True)
 tab = df.groupby('espece').count()
 tab = tab.sort_values(tab.columns[0],ascending = False).sample(NB_CLASS)
-
+tab.index = tab.index.astype(str)
 compte = pd.DataFrame(np.zeros((len(tab)+1,1)),columns = ['nombre'])
 
 nb_val = int(tab.file.min()*(1-args.ratio))