qyiSSKrSSKJs Jr SSKJr SSKJr SSK J r J r SSK J r Jr SSKJr SSKJr SS KJr \\"S S 9"S S \555rS /rg)N) BatchFeature)TorchvisionBackend)group_images_by_shapereorder_images)PILImageResamplingSizeDict)auto_docstring) TensorType)requires)torch)backendsc"\rSrSrSrSrSSS.r\Rr S\ SS\ S \ S \ S \ S \ S \ \ \ -S-S\ \ \ -S-S\ S-S\\-S-S\4SjrSS\R&S\ \R&S\ S\ \\\R&44SjjrSrg)UVDocImageProcessor"Tii)heightwidthimagesz torch.Tensor do_resizesize do_rescalerescale_factor do_normalize image_meanN image_stddisable_groupingreturn_tensorsreturnc [XS9up0nU R5H/unnURUXEXgU5nUSS2/SQSS2SS24nUX'M1 [X5nUR 5n[UU S9up0nU R5H?unnU(a-[ R "UURUR4SSS9nUUU'MA [UU 5n[UUS.U S/S9$) N)r)rbilinearTrmode align_corners) pixel_valuesoriginal_imagesr')data tensor_typeskip_tensor_conversion) ritemsrescale_and_normalizercopyF interpolaterrr)selfrrrrrrrrrrkwargsgrouped_imagesgrouped_images_indexprocessed_images_groupedshapestacked_imagesrescale_and_normalize_imagesr'interpolated_images_groupedr&s m/home/mande/repo/quber/.venv/lib/python3.13/site-packages/transformers/models/uvdoc/image_processing_uvdoc.py _preprocessUVDocImageProcessor._preprocess*s0EV/o,#% %3%9%9%; !E>!77 LV_N,Ay!Q,>?N.< $ + &<(66N'e$6;;=/D (;K0 ,')#%3%9%9%; !E>!""$++tzz)Bcg"2@ ' . &<&&ACWX ".?S&$5#6  predictionr'scalec[U5n[R"[U5URS9n/n[ U5HupgUR S:XaURS5nURUR5nURSSup[R"XUS-X4SSS 9n U RSSSS5n [R"X{SS 9n U RS5RSSS5n X-n U RS /S 9R[R SS S9n UR#SU 05 GM U$)aN Post-process document rectification predictions to convert them into rectified images. Args: prediction: Predicted 2D Bezier mesh coordinates, shape (B, 2, H, W) original_images: List of original input tensors, each of shape (C, H_i, W_i). Images may have different sizes. scale: Scaling factor for output images (default: 255.0) Returns: List of dictionaries containing rectified images. Each dictionary has: - "images": Rectified image tensor of shape (H, W, 3) with dtype torch.uint8 and BGR channel order (suitable for OpenCV visualization) )devicerrr Nr!r"Tr#)r%)dimsF)dtype non_blockingr-r)listr tensorfloatr@ enumeratendim unsqueezetor5r.r/permute grid_samplesqueezeflipuint8append)r0r=r'r> image_listresultsioriginal_imageoriginal_heightoriginal_widthupsampled_meshrearranged_mesh rectifiedimages r9#post_process_document_rectification7UVDocImageProcessor.post_process_document_rectification[sF&/*  U5\*2C2CD!*:!6 A""a'!/!9!9!!<+..z/@/@AN.<.B.B12.F +O]]q1u%%6" N-44Q1a@O nUYZI%%a(00Aq9EMEJJRDJ),,5;;TX],^E NNHe, -9"7<r<)go@)__name__ __module__ __qualname____firstlineno__rrrrBILINEARresamplerEboolr rGstrr rr:r Tensordictr\__static_attributes__r^r<r9rr"sJIC (D!**H/ ^$/ /  /  /  / / DK'$./ 4;&-/ +/ j(4//  / j 5LL5ell+5 5 d3 $% & 55r<r)r torch.nn.functionalnn functionalr.feature_extraction_utilsrimage_processing_backendsrimage_transformsrr image_utilsrr utilsr utils.genericr utils.import_utilsr r__all__r^r<r9rus\, 4;E7#'* :l,ll^ ! !r<