9i@SSKJrJrJrJr SSKrSSKJr SSKJ r SSK J r SSK J r SSKJr SSKJrJr SS KJr SS KJrJr SS KJr SS KJr \"5rS /r\R<"\R>\ R@S9"SS \55r!g))AnyDictOptionalUnionN) transforms) Pipelines)Model)NAFNetForImageDenoise) OutputKeys)InputPipeline) PIPELINES)ImageDenoisePreprocessor LoadImage)Tasks) get_loggerImageDenoisePipeline) module_namec^\rSrSrS S\\\4S\\4U4Sjjjr S\ S\ \\ 44Sjr SrS\ \\ 4S\ \\ 44S jrS\ \\ 4S\ \\ 44S jrS rU=r$) rmodel preprocessorc r>[TU]"SXS.UD6 URR5 URRUl[ R R5(a[ R"S5Ul O[ R"S5Ul [RS5 g)z use `model` and `preprocessor` to create a cv image denoise pipeline for prediction Args: model: model id on modelscope hub. )rrcudacpuzload image denoise model doneN) super__init__revalconfigtorchr is_availabledevice_deviceloggerinfo)selfrrkwargs __class__s n/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/pipelines/cv/image_denoise_pipeline.pyrImageDenoisePipeline.__init__sy JuJ6J jj'' :: " " $ $ <</DL <<.DL 34inputreturnc[R"U5n[R"[R"5/5nU"U5nSUR S5R UR50nU$)Nimgr)rconvert_to_imgrComposeToTensor unsqueezetor$)r'r-r0test_transformsresults r* preprocessImageDenoisePipeline.preprocess-s_&&u-$,,j.A.A.C-DEc"q),,T\\:; r,c *[R"U5nURSSup4[US-S5[US-S5peSnX5-XF-p[ U5GH5n [ U5GH!n X-n X-n XS- :aUOX<- nXS- :aU OXM- nUSS2SS2[SX- 5[ X-U-U52[SX- 5[ X-U-U524R 5n[SX- 5S:aUOSn[SX- 5S:aUOSn[ X-U-U5U:aUU-OUn[ X-U-U5U:aUU-OUnURRU5SSS2SS2UU2UU24USS2SS2XU-2XU-24'GM$ GM8 U$)Niroutputs) r! zeros_likeshapemaxrangemin contiguousr_inference_forward)r'r-outputihiw crop_rows crop_colsoverlapstep_hstep_wyxcrop_ycrop_xcrop_hcrop_w crop_framesh_startw_starth_endw_ends r* crop_process!ImageDenoisePipeline.crop_process4s !!%(RS!"29a0#bCi2C9"/y!A9%#$1}#4"+#$1}#4"+#Aq$'6+;%**-fo.G*L%M$'6+;%**-fo07/89;+=%=%=> ?Ijl  &)F,<%=%A'q%(F,<%=%A'q,/291:;=-?AC-D&(IK,/291:;=-?AC-D&(IK261N1N&2((12334a3:5=4A2Bq!VVO3vo-./)&"2 r,cSnSnU"URU5 [R"5 URUS5nSSS5 SU0$!,(df  SW0$=f)NcTU(aUR5 gUR5 gN)trainr)ris_trains r* set_phase/ImageDenoisePipeline.forward..set_phaseYs  r,Fr0 output_tensor)rr!no_gradrY)r'r-r`r_rFs r*forwardImageDenoisePipeline.forwardWs`  $**h' ]]_&&uU|4F ((_ ((s A A cUSRS5S-R5RSSS5R5R S5n[ R USS2SS2SSS240$)Nrbrr<uint8)squeezerpermutenumpyastyper OUTPUT_IMG)r'r- output_imgs r* postprocess ImageDenoisePipeline.postprocessfsiO,44Q7#=BBDLL q!UWVVG_ %%z!Q"*'=>>r,)r$r r])__name__ __module__ __qualname____firstlineno__rr strrrrr rrr8rYrdrq__static_attributes__ __classcell__)r)s@r*rrs EI53S895'(@A55&$sCx.!F )T#s(^ )S#X )?c3h?DcN??r,)"typingrrrrr! torchvisionrmodelscope.metainformodelscope.modelsr "modelscope.models.cv.image_denoiser modelscope.outputsr modelscope.pipelines.baser r modelscope.pipelines.builderrmodelscope.preprocessorsrrmodelscope.utils.constantrmodelscope.utils.loggerrr%__all__register_moduleimage_denoising image_denoiserrr,r*rsw-- ")#D)52H+.  ! "  y'>'>@Q?8Q?@Q?r,