From 4e1ba4f1ab2509aeea4b4958d9cc5b467f6661bf Mon Sep 17 00:00:00 2001
From: Paul Best <paul.best@lis-lab.fr>
Date: Wed, 28 Sep 2022 14:36:17 +0200
Subject: [PATCH] add globicephala in argparse

---
 run_CNN.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run_CNN.py b/run_CNN.py
index 93e441e..e87bb76 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)
-- 
GitLab