5
5
description = 'A program to detect bee buzzes using ML' ,
6
6
epilog = 'github.com/OSU-Bee-Lab/BuzzDetect'
7
7
)
8
+
8
9
subparsers = parser .add_subparsers (help = 'sub-command help' , dest = 'action' , required = True )
9
10
10
11
# analyze
11
12
parser_analyze = subparsers .add_parser ('analyze' , help = 'analyze something' )
12
13
parser_analyze .add_argument ('--modelname' , required = True , type = str )
13
14
parser_analyze .add_argument ('--dir_in' , required = False , default = "./audio_in" , type = str )
14
15
parser_analyze .add_argument ('--dir_out' , required = False , default = "./output" , type = str )
15
- parser_analyze .add_argument ('--frameLength' , required = False , default = 500 , type = int )
16
- parser_analyze .add_argument ('--frameHop' , required = False , default = 250 , type = int )
16
+ parser_analyze .add_argument ('--chunkLength' , required = False , default = 1 , type = float )
17
+ parser_analyze .add_argument ('--frameLength' , required = False , default = 1000 , type = int )
18
+ parser_analyze .add_argument ('--frameHop' , required = False , default = 750 , type = int )
17
19
18
20
# train
19
21
parser_train = subparsers .add_parser ('train' , help = 'train a new model' )
40
42
from buzzcode .analyze import *
41
43
42
44
print ("analyzing audio in " + args .dir_in + " with model " + args .modelname )
43
- analyze_batch ( model_name = args .modelname , directory_in = args .dir_in , directory_out = args .dir_out , frameLength = args .frameLength , frameHop = args .frameHop )
45
+ analyze_mp3_batch ( modelname = args .modelname , directory_in = args .dir_in , directory_out = args .dir_out , frameLength = args .frameLength , frameHop = args .frameHop , chunklength_hr = args . chunkLength )
0 commit comments