Optimized loop and bugfix in SDE (NVIDIA#5573)

- Fixed bug with loading custom data attributes from JSON in Speech Data Explorer Signed-off-by: George Zelenfroynd <[email protected]>
virajkarandikar · Dec 8, 2022 · f091999 · f091999
1 parent 584ca56
commit f091999
Showing 1 changed file with 9 additions and 13 deletions.
diff --git a/tools/speech_data_explorer/data_explorer.py b/tools/speech_data_explorer/data_explorer.py
@@ -298,19 +298,15 @@ def append_data(
                     data[-1]['I'] = measures['insertions']
                     data[-1]['D'] = measures['deletions']
                     data[-1]['D-I'] = measures['deletions'] - measures['insertions']
-                else:
-                    for k in item:
-                        if k not in data[-1]:
-                            data[-1][k] = item[k]
-            if estimate_audio:
-                filepath = absolute_audio_filepath(item['audio_filepath'], data_filename)
-                signal, sr = librosa.load(path=filepath, sr=None)
-                bw = eval_bandwidth(signal, sr)
-                item['freq_bandwidth'] = int(bw)
-                item['level_db'] = 20 * np.log10(np.max(np.abs(signal)))
-            for k in item:
-                if k not in data[-1]:
-                    data[-1][k] = item[k]
+                if estimate_audio:
+                    filepath = absolute_audio_filepath(item['audio_filepath'], data_filename)
+                    signal, sr = librosa.load(path=filepath, sr=None)
+                    bw = eval_bandwidth(signal, sr)
+                    item['freq_bandwidth'] = int(bw)
+                    item['level_db'] = 20 * np.log10(np.max(np.abs(signal)))
+                for k in item:
+                    if k not in data[-1]:
+                        data[-1][k] = item[k]
 
             vocabulary_data = [{'word': word, 'count': vocabulary[word]} for word in vocabulary]
             return (