update

.gitignore

@@ -12,7 +12,7 @@
 /docs/
 /dotenv/
 /examples/**/*.wav
-/trained_models/
+/trained_models*/
 /temp/
 
 **/*.csv

examples/sound_classification_by_lstm/run.sh

@@ -187,8 +187,7 @@ if [ ${stage} -le 6 ] && [ ${stop_stage} -ge 6 ]; then
   cp "${evaluation_file}" "${final_model_dir}/evaluation.xlsx"
   cp "${evaluation_onnx_file}" "${final_model_dir}/evaluation-onnx.xlsx"
 
-  cp "${onnx_preprocess_file}" "${onnx_preprocess_file}"
-  cp "${onnx_model_file}" "${onnx_model_file}"
+  cp "${onnx_model_file}" "${final_model_dir}"
 
   cp "${file_dir}/trace_model.zip" "${final_model_dir}/trace_model.zip"
   cp "${file_dir}/trace_quant_model.zip" "${final_model_dir}/trace_quant_model.zip"

tabs/event_tab.py

@@ -198,16 +198,10 @@ def when_click_event_button_by_lstm_model(audio_t,
             waveform_ = inputs[:, begin:end]
             if waveform_.shape[-1] < win_size:
                 break
-            spec = wave_to_mel_spectrogram(waveform_) + 1e-6
-            spec = spec.log()
-            # shape = [b, f, t]
-            spec = spec.transpose(1, 2)
-            # shape = [b, t, f]
-            sub_inputs = spec
 
             # infer
             input_feed = {
-                "inputs": sub_inputs.numpy(),
+                "inputs": waveform_.numpy(),
                 "h": h,
                 "c": c,
             }