diff --git a/run_CNN.py b/run_CNN.py index 93e441ed60734ace049ca4549c130ac3af241793..e87bb767f759b8b3e35306387f9db39fcaf03601 100644 --- a/run_CNN.py +++ b/run_CNN.py @@ -10,7 +10,7 @@ import argparse parser = argparse.ArgumentParser(description="Run this script to use a CNN for the detection of cetacean vocalizations on a folder of audio files.") parser.add_argument('audio_folder', type=str, help='Path of the folder with audio files to process') -parser.add_argument('specie', type=str, help='Target specie to detect', choices=['megaptera', 'delphinid', 'orcinus', 'physeter', 'balaenoptera']) +parser.add_argument('specie', type=str, help='Target specie to detect', choices=['megaptera', 'delphinid', 'orcinus', 'physeter', 'balaenoptera', 'globicephala']) parser.add_argument('-lensample', type=float, help='Length of the signal for each sample (in seconds)', default=5), parser.add_argument('-batch_size', type=int, help='Amount of samples to process at a time (usefull for parallel computation using a GPU)', default=32), parser.add_argument('-channel', type=int, help='Channel of the audio file to use in the model inference (starting from 0)', default=0)