Skip to content

Commit 056b2d6

Browse files
committed
Standardize network conversion
1 parent f62eb27 commit 056b2d6

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

nndetector_live.m

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,8 @@ function nndetector_live(varargin)
7575
end
7676
end
7777

78-
load(net_file,'net');
78+
%load(net_file,'net');
79+
net=nndetector_live_load_net(net_file);
7980
network=nndetector_live_convert_net(net);
8081

8182
if ~isempty(manual_threshold)

nndetector_live_load_net.m

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
function net=nndetector_live_load_net(net_file)
2+
load(net_file);
3+
net.userdata.fft_size = fft_size;
4+
net.userdata.win_size = fft_size;
5+
net.userdata.fft_time_shift = fft_time_shift;
6+
net.userdata.amp_scaling = 'linear';
7+
net.userdata.inp_scaling = 'zscore';
8+
net.userdata.freq_range = [1000 8000];
9+
net.userdata.freq_range_ds = freq_range_ds;
10+
net.userdata.trigger_thresholds = trigger_thresholds;
11+
net.userdata.time_window_steps = time_window_steps;
12+
net.userdata.time_window = time_window_steps;
13+
end

0 commit comments

Comments
 (0)