Update pitch/inference.py

pitch/inference.py

@@ -55,7 +55,20 @@ def compute_f0_voice(filename, device):
     fmax = 1000
     model = "full"
     batch_size = 512
-    pitch = crepe.predict(
+    # pitch = crepe.predict(
+    #     audio,
+    #     sr,
+    #     hop_length,
+    #     fmin,
+    #     fmax,
+    #     model,
+    #     batch_size=batch_size,
+    #     device=device,
+    #     return_periodicity=False,
+    # )
+    # pitch = crepe.filter.mean(pitch, 3)
+    # pitch = pitch.squeeze(0)
+    pitch, periodicity = crepe.predict(
         audio,
         sr,
         hop_length,
@@ -64,9 +77,12 @@ def compute_f0_voice(filename, device):
         model,
         batch_size=batch_size,
         device=device,
-        return_periodicity=False,
+        return_periodicity=True,
     )
-    pitch = crepe.filter.mean(pitch, 3)
+    # CREPE was not trained on silent audio. some error on silent need filter.pitPath
+    periodicity = crepe.filter.median(periodicity, 7)
+    pitch = crepe.filter.mean(pitch, 5)
+    pitch[periodicity < 0.5] = 0
     pitch = pitch.squeeze(0)
     return pitch
 
@@ -128,7 +144,8 @@ if __name__ == "__main__":
     print(args.pit)
 
     device = "cuda" if torch.cuda.is_available() else "cpu"
-    pitch = compute_f0_sing(args.wav, device)
+    # pitch = compute_f0_sing(args.wav, device)
+    pitch = compute_f0_voice(args.wav, device)
     save_csv_pitch(pitch, args.pit)
     # tmp = load_csv_pitch(args.pit)
     # save_csv_pitch(tmp, "tmp.csv")