Skip to content

Commit 2390323

Browse files
committed
update buzzdetect.py to work with new analysis
1 parent 7338d43 commit 2390323

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

buzzdetect.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,15 +5,17 @@
55
description='A program to detect bee buzzes using ML',
66
epilog='github.com/OSU-Bee-Lab/BuzzDetect'
77
)
8+
89
subparsers = parser.add_subparsers(help='sub-command help', dest='action', required = True)
910

1011
# analyze
1112
parser_analyze = subparsers.add_parser('analyze', help = 'analyze something')
1213
parser_analyze.add_argument('--modelname', required=True, type = str)
1314
parser_analyze.add_argument('--dir_in', required=False, default = "./audio_in", type = str)
1415
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)
1719

1820
# train
1921
parser_train = subparsers.add_parser('train', help = 'train a new model')
@@ -40,4 +42,4 @@
4042
from buzzcode.analyze import *
4143

4244
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

Comments
 (0)