9i<SSKrSSKJr SSKJrJr SSKrSSK r SSK J r SSK J r Jr SSKJr SSKJr SSKJr SSKJrJr SSKJr SS KJr SS KJr SS KJrJ r SS K!J"r" \""5r#\RH"\ RJ\RLS 9"SS\55r'g)N)AnyDict)collatescatter) Pipelines)VisionMiddlewareModel) OutputKeys)InputPipeline) PIPELINES) LoadImage)Config) ModelFileTasks) get_logger) module_namec^\rSrSrS\4U4SjjrS\S\\\44Sjr S\\\4S\\\44Sjr S\\\4S\\\44S jr S r U=r $) VisionMiddlewarePipelinemodelc T>[TU]"SSU0UD6 URR5UlURR 5 [ R "[ R"S5[ R"5[ R"/SQ/SQS9/5Ul g)zp use `model` to create a vision middleware pipeline for prediction Args: model: model id on modelscope hub. r)r)g ףp= ?gv/?gCl?)gZd;O?gy&1?g?)meanstdN) super__init__rcudaeval transformsComposeResizeToTensor Normalize transform)selfrkwargs __class__s r/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/pipelines/cv/vision_middleware_pipeline.pyr!VisionMiddlewarePipeline.__init__s /u//ZZ__&  #++   j )    !  *0E G-  inputreturncD[R"U5nURU5n[U/SS9n[ UR R 55R(a;[U[ UR R 55R/5SnU$)N)samples_per_gpur) r convert_to_imgr%rnextr parametersis_cudardevice)r&r,imgdatas r) preprocess#VisionMiddlewarePipeline.preprocess/sz&&u-~~c"vq1  %%' ( 0 04$tzz'<'<'>"?"F"F!GHKD r+c[R"5 URUSS9nUsSSS5 $!,(df  g=f)Nzseg-voc) task_name)torchno_gradr)r&r,resultss r)forward VisionMiddlewarePipeline.forward;s* ]]_jj)jrbsv  +*)H)52.*6.   668&x&8&r+