纸张矫正算法笔记
DocUNet
UDoc-GAN
DewarpNet
docuwarp
pstwh/docuwarp: Docuwarp is a Python library for unwarping documents
修改后代码:
envs\py311\Lib\site-packages\docuwarp\unwarp.py
def inference(self, image: Image.Image) -> Image.Image: resized_input, original_input, original_size = self.prepare_input(image) points, _ = self.session.run(None, {"input": resized_input.astype(np.float16)}) unwarped = self.bilinear_unwarping.run( None, { "warped_img": original_input.astype(np.float32),