o i5@srdZddlZddlZddlZddlZddlZddlZddlm Z ddZ ddZ e dkr7e ej d ddSdS) z4End-to-end speech recognition model decoding script.N) strtoboolcCstjdtjtjd}|jdddd|jdddd|jd dd d|jd td d d|jddddd|jdtdddgdd|jdtddd|jdtddd|jddtdd d|jd!tdd"d|jd#td$d%d|jd&d'd'd(gd)d*|jd+td,d-|jd.tdd/d0|jd1tdd2d0|jd3td$d4d|jd5tddd6gd7d|jd8dtd9d:|jd;tddd|jd?td@dAd|jdBtd@dCd|jdDtd@dEd|jdFtd@dGd|jdHtdIdJdK|jdLtd dMd|jdNtdOgdPdQd|jdRtddSd|jdTtd6dUd|jdVtd6dWd|jdXtdYdZd|jd[td\d]d|jd^td6d_d|jd`t daddbdc|jddtdedfd|jdgtd$dhd|jditd$djd|jdktd$dld|jdmtd$dnd|jdotd$dpd|jdqtdrdsd|jdttd$dud|jdvtdrdwd|jdxtdydzd{d|jd|td$d}d~gdd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdtddd|jdddd|jdtdddgdd|jdt ddd|jdt ddd|S)zGet default arguments.zNTranscribe text from speech using a speech recognition model on one CPU or GPU) descriptionconfig_file_parser_classformatter_classz--configTzConfig file path)is_config_filehelpz --config2zBSecond config file path that overwrites the settings in `--config`z --config3zQThird config file path that overwrites the settings in `--config` and `--config2`z--ngpurzNumber of GPUs)typedefaultrz--dtype)float16float32float64r z,Float precision (only available in --api v2))choicesr rz --backendchainerpytorchzBackend library)rr r rz --debugmode Debugmodez--seedz Random seedz --verbosez-VzVerbose optionz --batchsizez9Batch size for beam search (0: means no batch processing)z--preprocess-confNz-The configuration file for the pre-processingz--apiv1v2zBeam search APIs v1: Default API. It only supports the ASRInterface.recognize method and DefaultRNNLM. v2: Experimental API. It supports any models that implements ScorerInterface.)r r rz --recog-jsonz#Filename of recognition data (json))rrz--result-labelz$Filename of result label data (json))rrequiredrz--modelzModel file parameters to readz --model-confzModel config filez --num-spkrsz Number of speakers in the speechz --num-encsz Number of encoders in the model.)r rrz--nbestzOutput N-best hypothesesz --beam-sizez Beam sizez --penaltygzIncertion penaltyz --maxlenratioaSInput length ratio to obtain max output length. If maxlenratio=0.0 (default), it uses a end-detect function to automatically find maximum hypothesis lengths. If maxlenratio<0.0, its absolute value is interpreted as a constant max output lengthz --minlenratioz.Input length ratio to obtain min output lengthz --ctc-weightzCTC weight in joint decodingz--weights-ctc-decappendzPctc weight assigned to each encoder during decoding.[in multi-encoder mode only])ractionrz--ctc-window-marginaUse CTC window with margin parameter to accelerate CTC/attention decoding especially on GPU. Smaller magin makes decoding faster, but may increase search errors. If margin=0 (default), this function is disabledz --search-typer )r nsctsdalsdmaesa5Type of beam search implementation to use during inference. Can be either: default beam search ("default"), N-Step Constrained beam search ("nsc"), Time-Synchronous Decoding ("tsd"), Alignment-Length Synchronous Decoding ("alsd") or modified Adaptive Expansion Search ("maes").z--nstepzpNumber of expansion steps allowed in NSC beam search or mAES (nstep > 0 for NSC and nstep > 1 for mAES).z--prefix-alphaz%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s)levelformatrzSkip DEBUG/INFO messagesCUDA_VISIBLE_DEVICESNz CUDA_VISIBLE_DEVICES is not set.,z/#gpus is not matched with CUDA_VISIBLE_DEVICES.z!The program only supports ngpu=1.zpython path = PYTHONPATHz(None)zset random seed = %dzTIt seems that both --rnnlm and --word-rnnlm are specified. Please use either option.z backend = r)recogrr)recog_v2r z `--dtype z#` is only available with `--api v2`z --num-encs z! > 1 is not supported in --api v2z'Only chainer and pytorch are supported.zOnly pytorch is supported.)%r2 parse_argsngpudtype ValueErrorverboselogging basicConfigINFODEBUGWARNwarningosenvirongetlenspliterrorsysexitinforandomseednprnnlm word_rnnlmbackend num_spkrsespnet.asr.chainer_backend.asrr8num_encsapi espnet.asr.pytorch_backend.recogr9espnet.asr.pytorch_backend.asrNotImplementedError"espnet.asr.pytorch_backend.asr_mix)argsr/cvdr8r9r0r0r1main8s                                   r^__main__r)__doc__r?rErNrKr%numpyrPespnet.utils.cli_utilsrr2r^__name__argvr0r0r0r1s $b