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

Update file get_time_freq_detection.py

parent 37d91414
No related branches found
No related tags found
No related merge requests found
...@@ -47,9 +47,9 @@ def process_annotations(annotations_folder, duration, outdir, sr, names): ...@@ -47,9 +47,9 @@ def process_annotations(annotations_folder, duration, outdir, sr, names):
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description='Collect detections and return a complete dataframe') parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter, description='Collect detections and return a complete dataframe')
parser.add_argument('path_to_data', type=arg_directory, help='Path of the folder that contains the .txt files', required=True) parser.add_argument('path_to_data', type=arg_directory, help='Path of the folder that contains the .txt files')
parser.add_argument('directory', type=arg_directory, help='Directory where the dataframe will be stored', required=True) parser.add_argument('directory', type=arg_directory, help='Directory where the dataframe will be stored')
parser.add_argument('names', type=str, help='path to YOLOv5 custom_data.yaml file', required=True) parser.add_argument('names', type=str, help='path to YOLOv5 custom_data.yaml file')
parser.add_argument('-s','--sr', type=int, help='Sampling Rate of the spectrogram', required=True) parser.add_argument('-s','--sr', type=int, help='Sampling Rate of the spectrogram', required=True)
parser.add_argument('--duration', type=int, help='Duration of the spectrogram', default=8) parser.add_argument('--duration', type=int, help='Duration of the spectrogram', default=8)
args = parser.parse_args() args = parser.parse_args()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment