Update utils/pitch_ld_extraction.py

utils/pitch_ld_extraction.py

@@ -60,9 +60,12 @@ def REAPER_F0(wav_path, sr=24000, frame_period=0.01):  # frame_period s
         cmd = f'reaper-tmp/REAPER/build/reaper -i {wav_path} -f {wav_path}.f0 -e {frame_period} -x 1000 -m 65 -a'
         os.system(cmd)
     f0 = []
-    with open(f'{wav_path}.f0', 'r') as rf:
-        for line in rf.readlines()[7:]:
-            f0.append(float(line.split()[2]))
+    try:
+        with open(f'{wav_path}.f0', 'r') as rf:
+            for line in rf.readlines()[7:]:
+                f0.append(float(line.split()[2]))
+    except FileNotFoundError as e:
+        return None
 
     cmd = f'rm -f {wav_path}.f0'
     os.system(cmd)
@@ -159,7 +162,8 @@ def compute_pitch(wav_path: str, pitch_path: str=None, frame_period=0.01):
         compute_median.append(f0)
         # Compute pitch using REAPER algorithm
         f0 = REAPER_F0(wav_path, sr=fs, frame_period=frame_period)
-        compute_median.append(f0)
+        if f0 is not None:
+            compute_median.append(f0)
 
         # Compute median F0
         compute_median = pad_arrays(compute_median, f0_std_len)
@@ -249,4 +253,4 @@ if __name__ == '__main__':
     parser.add_argument('--n_cpu', type=int, default=1)
 
     args = parser.parse_args()
-    main(args)
+    main(args)