9i S SK Jr S SKJr S SKJrJr S SKJrJ r S SK
J r S SK J
r
S SKJr \R " \R" \R$ S9 " S S
\5 5 rg ) )nn)Models)MODELS
TorchModel)RectifiedLinearSigmoid)AffineTransform) UniDeepFsmn)Tasks) module_namec H ^ \ rS rSrSr SS\4U 4S jjjrS rS rSr U =r
$ ) DfsmnAns zDenoise model with DFSMN.
Args:
model_dir (str): the model path.
fsmn_depth (int): the depth of deepfsmn
lorder (int):
model_dirc J > [ TU ] " U/UQ70 UD6 X0l [ SS5 U l [ SS5 U l [ U5 Vs/ s H n[ SSUS5 PM nn[ R " U6 U l [ SS5 U l [ SS5 U l g s snf )Nx i )super__init__lorderr linear1r reluranger
r
Sequentialdeepfsmnlinear2r sig) selfr
fsmn_depthr argskwargsirepeats __class__s g/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/audio/ans/denoise_net.pyr DfsmnAns.__init__ s 4T4V4 &sC0 #C- 8=j8I
8I1KS&# .8I
w/
&sC0 3$
s B c U R U5 nU R U5 nU R U5 nU R U5 nU R U5 nU$ )z
Args:
input: fbank feature [batch_size,number_of_frame,feature_dimension]
Returns:
mask value [batch_size, number_of_frame, FFT_size/2+1]
)r r r r r )r inputx1x2x3x4x5s r% forwardDfsmnAns.forward/ sM \\%
YYr]
]]2
\\"
XXb\ c H SnUS-
nXR R 5 -
nXR R 5 -
nU R H nXR 5 -
nM XR R 5 -
nXR
R 5 -
nUS-
nU$ )N z
z
)r
to_kaldi_nnetr r r r )r re_strdfsmns r% r3 DfsmnAns.to_kaldi_nnet> s * ,,,,..))))++]]E ))+ +F #,,,,..((((**+
r0 )r r r r r r ) ) __name__
__module__ __qualname____firstlineno____doc__strr r. r3 __static_attributes__
__classcell__)r$ s @r% r r s1 % % %"
r0 r N)torchr modelscope.metainfor modelscope.modelsr r .modelscope.models.audio.ans.layers.activationsr r 3modelscope.models.audio.ans.layers.affine_transformr 0modelscope.models.audio.ans.layers.uni_deep_fsmnr
modelscope.utils.constantr register_moduleacoustic_noise_suppressionspeech_dfsmn_ansr r0 r% rL sX & 0E O H + $$&2I2IK4z 4K4r0