猪大大Andy

摘要： 1 import os 2 import torch 3 import numpy as np 4 from Unet import UNET 5 os.environ["CUDA_VISIBLE_DEVICE"] = "" 6 7 def main(): 8 demo = Demo(model_p 阅读全文

摘要： 1 前言： 1.1 我用UNet模型来检测表格的行列线，模型（基于pytorch）训练好之后预测准确，想转换成ONNX模型来部署，结果遇到了转换后的ONNX模型推理结果有误的问题 2 问题排查： 2.1 输入图片、预处理以及后处理是否一致 pytorch模型下推理输入的是图片的tensor，输出保存 阅读全文