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

Update get_train_val_YOLO.py

parent f29bc82c
Branches
No related tags found
No related merge requests found
...@@ -10,7 +10,7 @@ import ipdb ...@@ -10,7 +10,7 @@ import ipdb
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description='TODO') parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description='TODO')
parser.add_argument('-r','--ratio', type=float, default=0.7,help = 'Train Ratio (val = 1 - 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 .txt (ending with labels/)',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 = 'Directory to wich spectrogram and the .txt files will be stored (different from -p)',required=True) parser.add_argument('-d','--direction', type=str, help = 'Directory to wich spectrogram and .txt files will be stored (different from -p)',required=True)
args = parser.parse_args() args = parser.parse_args()
path = str(args.path_to_data) path = str(args.path_to_data)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment