Fixed main and merging this to r1.20 (NVIDIA#7127)

* Fixed main and merging this to r1.20 Signed-off-by: Taejin Park <[email protected]> * Update vad_utils.py Signed-off-by: He Huang (Steve) <[email protected]> --------- Signed-off-by: Taejin Park <[email protected]> Signed-off-by: He Huang (Steve) <[email protected]> Co-authored-by: He Huang (Steve) <[email protected]>
zhehuaichen · Oct 4, 2023 · 33ca0d7 · 33ca0d7
1 parent 0e3a545
commit 33ca0d7
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/nemo/collections/asr/parts/utils/vad_utils.py b/nemo/collections/asr/parts/utils/vad_utils.py
@@ -275,7 +275,9 @@ def generate_overlap_vad_seq(
     if out_dir:
         overlap_out_dir = out_dir
     else:
-        overlap_out_dir = frame_pred_dir + "/overlap_smoothing_output" + "_" + smoothing_method + "_" + str(overlap)
+        overlap_out_dir = os.path.join(
+            frame_pred_dir, "overlap_smoothing_output" + "_" + smoothing_method + "_" + str(overlap)
+        )
 
     if not os.path.exists(overlap_out_dir):
         os.mkdir(overlap_out_dir)
@@ -730,9 +732,9 @@ def generate_vad_segment_table(
     vad_pred_filepath_list = [os.path.join(vad_pred_dir, x) for x in os.listdir(vad_pred_dir) if x.endswith(suffixes)]
 
     if not out_dir:
-        out_dir_name = "seg_output_"
+        out_dir_name = "seg_output"
         for key in postprocessing_params:
-            out_dir_name = out_dir_name + str(key) + str(postprocessing_params[key]) + "-"
+            out_dir_name = out_dir_name + "-" + str(key) + str(postprocessing_params[key])
 
         out_dir = os.path.join(vad_pred_dir, out_dir_name)
 
@@ -1049,11 +1051,11 @@ def plot(
     else:
         label = None
 
-    if label:
+    if label is not None:
         ax2.plot(np.arange(len_pred) * unit_frame_len, label, 'r', label='label')
-    if pred_snippet:
+    if pred_snippet is not None:
         ax2.plot(np.arange(len_pred) * unit_frame_len, pred_snippet, 'b', label='pred')
-    if frame_snippet:
+    if frame_snippet is not None:
         ax2.plot(np.arange(len_pred) * unit_frame_len, frame_snippet, 'g--', label='speech prob')
 
     ax2.tick_params(axis='y', labelcolor='r')