Update get_train_val_YOLO.py

get_train_val_YOLO.py

@@ -9,8 +9,8 @@ import ipdb
 
 parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description='RATIO')
 parser.add_argument('-r','--ratio', type=float, default=0.7,help = 'Train Ratio (val = 1 - ratio)')
-parser.add_argument('-p','--path_to_data', type=str, help = 'Path of the folder that contain the recodings',required=True)
-parser.add_argument('-d','--direction', type=str, help = 'Direction of the folder to export the spectrogram and the .txt files',required=True)
+parser.add_argument('-p','--path_to_data', type=str, help = 'Path of the folder that contain the .txt (ending with labels/)',required=True)
+parser.add_argument('-d','--direction', type=str, help = 'Direction of the folder to export the spectrogram and the .txt files (different from -p)',required=True)
 args = parser.parse_args()
 
 path = str(args.path_to_data)
@@ -19,7 +19,6 @@ direction = str(args.direction)
 files = os.listdir(path)
 
 files = pd.DataFrame(files, columns = ['file'])
-files['annot'] = 'None'
 
 df = pd.DataFrame(columns = ['file','espece','x','y','w','h',])
 for i in tqdm(os.listdir(path)):