avoid nan error

TheCacophonyProject · Jun 10, 2021 · 1be45ef · 1be45ef
1 parent 1230645
commit 1be45ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Melt/identify_species.py b/Melt/identify_species.py
@@ -144,8 +144,10 @@ def identify_species(recording, metadata, models):
         start = limit - slices_per_sample
         start_times.append(start / slices_per_second)
         sample = npspec[:, start:limit]
+
         sample = librosa.amplitude_to_db(sample, ref=np.max)
-        sample = sample / abs(sample.min()) + 1.0
+        if sample.min() != 0:
+            sample = sample / abs(sample.min()) + 1.0
         samples.append(sample.reshape(sample.shape + (1,)))
     samples = np.array(samples)