[cli] remove output_file para for the diarization task

README.md

@@ -24,7 +24,7 @@ pip install git+https://github.com/wenet-e2e/wespeaker.git
 ``` sh
 $ wespeaker --task embedding --audio_file audio.wav --output_file embedding.txt
 $ wespeaker --task similarity --audio_file audio.wav --audio_file2 audio2.wav
-$ wespeaker --task diarization --audio_file audio.wav --output_file diarization.txt  # TODO
+$ wespeaker --task diarization --audio_file audio.wav  # TODO
 ```
 
 **Python programming usage**:

docs/python_package.md

@@ -20,7 +20,7 @@ pip install -e .
 ``` sh
 $ wespeaker --task embedding --audio_file audio.wav --output_file embedding.txt
 $ wespeaker --task similarity --audio_file audio.wav --audio_file2 audio2.wav
-$ wespeaker --task diarization --audio_file audio.wav --output_file diarization.txt  # TODO
+$ wespeaker --task diarization --audio_file audio.wav  # TODO
 ```
 
 You can specify the following parameters. (use `-h` for details)
@@ -34,7 +34,7 @@ You can specify the following parameters. (use `-h` for details)
 * `--audio_file2`: input audio file2 path, spicifically for the similarity task
 * `--resample_rate`: resample rate (default: 16000)
 * `--vad`: apply vad or not for the input audios (default: true)
-* `--output_file`: output file to save speaker embedding or diarization result
+* `--output_file`: output file to save speaker embedding
 
 ## Python Programming Usage
 

wespeaker/cli/speaker.py

@@ -142,9 +142,8 @@ def get_args():
     parser.add_argument('--vad',
                         action='store_true',
                         help='whether to do VAD or not')
-    parser.add_argument(
-        '--output_file',
-        help='output file to save speaker embedding or diarization result')
+    parser.add_argument('--output_file',
+                        help='output file to save speaker embedding')
     args = parser.parse_args()
     return args
 
@@ -166,10 +165,8 @@ def main():
     elif args.task == 'diarization':
         # TODO(Chengdong Liang): Add diarization surport
         diar_result = model.diarize(args.audio_file)
-        with open(args.output_file, "w") as fout:
-            for (start, end, spkid) in diar_result:
-                fout.write("{:.3f}\t{:.3f}\t{:d}\n".format(start, end, spkid))
-        print('Succeed, see {}'.format(args.output_file))
+        for (start, end, spkid) in diar_result:
+            print("{:.3f}\t{:.3f}\t{:d}".format(start, end, spkid))
     else:
         print('Unsupported task {}'.format(args.task))
         sys.exit(-1)