Merge pull request #69 from alibaba-damo-academy/dev_lhn

Dev lhn

funasr/utils/asr_utils.py

@@ -58,7 +58,7 @@ def type_checking(audio_in: Union[str, bytes],
     if r_recog_type is None and audio_in is not None:
         # audio_in is wav, recog_type is wav_file
         if os.path.isfile(audio_in):
-            audio_type = os.path.basename(audio_in).split(".")[1].lower()
+            audio_type = os.path.basename(audio_in).split(".")[-1].lower()
             if audio_type in SUPPORT_AUDIO_TYPE_SETS:
                 r_recog_type = 'wav'
                 r_audio_format = 'wav'
@@ -128,7 +128,7 @@ def get_sr_from_bytes(wav: bytes):
 def get_sr_from_wav(fname: str):
     fs = None
     if os.path.isfile(fname):
-        audio_type = os.path.basename(fname).split(".")[1].lower()
+        audio_type = os.path.basename(fname).split(".")[-1].lower()
         if audio_type in SUPPORT_AUDIO_TYPE_SETS:
             if audio_type == "pcm":
                 fs = None
@@ -140,7 +140,7 @@ def get_sr_from_wav(fname: str):
         for file in dir_files:
             file_path = os.path.join(fname, file)
             if os.path.isfile(file_path):
-                audio_type = os.path.basename(file_path).split(".")[1].lower()
+                audio_type = os.path.basename(file_path).split(".")[-1].lower()
                 if audio_type in SUPPORT_AUDIO_TYPE_SETS:
                     fs = get_sr_from_wav(file_path)
             elif os.path.isdir(file_path):
@@ -158,7 +158,7 @@ def find_file_by_ends(dir_path: str, ends: str):
         file_path = os.path.join(dir_path, file)
         if os.path.isfile(file_path):
             if ends == ".wav" or ends == ".WAV":
-                audio_type = os.path.basename(file_path).split(".")[1].lower()
+                audio_type = os.path.basename(file_path).split(".")[-1].lower()
                 if audio_type in SUPPORT_AUDIO_TYPE_SETS:
                     return True
                 else:
@@ -178,7 +178,7 @@ def recursion_dir_all_wav(wav_list, dir_path: str) -> List[str]:
     for file in dir_files:
         file_path = os.path.join(dir_path, file)
         if os.path.isfile(file_path):
-            audio_type = os.path.basename(file_path).split(".")[1].lower()
+            audio_type = os.path.basename(file_path).split(".")[-1].lower()
             if audio_type in SUPPORT_AUDIO_TYPE_SETS:
                 wav_list.append(file_path)
         elif os.path.isdir(file_path):