| | |
|---|
| | | type=str, |
|---|
| | | default=None, |
|---|
| | | help="audio_in") |
|---|
| | | parser.add_argument("--audio_fs", |
|---|
| | | type=int, |
|---|
| | | default=16000, |
|---|
| | | help="audio_fs") |
|---|
| | | parser.add_argument("--send_without_sleep", |
|---|
| | | action="store_true", |
|---|
| | | default=True, |
|---|
| | |
|---|
| | | hotword_msg=json.dumps(fst_dict) |
|---|
| | | print (hotword_msg) |
|---|
| | | |
|---|
| | | sample_rate = 16000 |
|---|
| | | sample_rate = args.audio_fs |
|---|
| | | wav_format = "pcm" |
|---|
| | | use_itn=True |
|---|
| | | if args.use_itn == 0: |
|---|
| | |
|---|
| | | if wav_path.endswith(".pcm"): |
|---|
| | | with open(wav_path, "rb") as f: |
|---|
| | | audio_bytes = f.read() |
|---|
| | | elif wav_path.endswith(".wav"): |
|---|
| | | import wave |
|---|
| | | with wave.open(wav_path, "rb") as wav_file: |
|---|
| | | params = wav_file.getparams() |
|---|
| | | sample_rate = wav_file.getframerate() |
|---|
| | | frames = wav_file.readframes(wav_file.getnframes()) |
|---|
| | | audio_bytes = bytes(frames) |
|---|
| | | else: |
|---|
| | | wav_format = "others" |
|---|
| | | with open(wav_path, "rb") as f: |
|---|
| | | audio_bytes = f.read() |
|---|
| | | |
|---|
| | | # stride = int(args.chunk_size/1000*16000*2) |
|---|
| | | stride = int(60 * args.chunk_size[1] / args.chunk_interval / 1000 * 16000 * 2) |
|---|
| | | stride = int(60 * args.chunk_size[1] / args.chunk_interval / 1000 * sample_rate * 2) |
|---|
| | | chunk_num = (len(audio_bytes) - 1) // stride + 1 |
|---|
| | | # print(stride) |
|---|
| | | |
|---|
| | |
|---|
| | | wav_name = meg.get("wav_name", "demo") |
|---|
| | | text = meg["text"] |
|---|
| | | timestamp="" |
|---|
| | | offline_msg_done = meg.get("is_final", False) |
|---|
| | | if "timestamp" in meg: |
|---|
| | | timestamp = meg["timestamp"] |
|---|
| | | |
|---|
| | |
|---|
| | | else: |
|---|
| | | text_write_line = "{}\t{}\n".format(wav_name, text) |
|---|
| | | ibest_writer.write(text_write_line) |
|---|
| | | |
|---|
| | | |
|---|
| | | if 'mode' not in meg: |
|---|
| | | continue |
|---|
| | | if meg["mode"] == "online": |
|---|
| | | text_print += "{}".format(text) |
|---|
| | | text_print = text_print[-args.words_max_print:] |
|---|
| | |
|---|
| | | text_print = text_print[-args.words_max_print:] |
|---|
| | | os.system('clear') |
|---|
| | | print("\rpid" + str(id) + ": " + text_print) |
|---|
| | | offline_msg_done=True |
|---|
| | | # offline_msg_done=True |
|---|
| | | |
|---|
| | | except Exception as e: |
|---|
| | | print("Exception:", e) |
|---|
