项目架构调整,模型全部采用接口调用
This commit is contained in:
@@ -1,4 +1,7 @@
|
||||
import os.path
|
||||
|
||||
from onnxruntime import InferenceSession
|
||||
|
||||
DOC_TR = InferenceSession("model/dewarp_model/doc_tr_pp.onnx",
|
||||
providers=["CUDAExecutionProvider"], provider_options=[{"device_id": 0}])
|
||||
MODEL_PATH = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))),
|
||||
'model', 'dewarp_model', 'doc_tr_pp.onnx')
|
||||
DOC_TR = InferenceSession(MODEL_PATH, providers=['CUDAExecutionProvider'], provider_options=[{'device_id': 0}])
|
||||
|
||||
@@ -11,10 +11,10 @@ def dewarp_image(image):
|
||||
y = to_tensor(image)
|
||||
|
||||
img = np.transpose(img, (2, 0, 1))
|
||||
bm = DOC_TR.run(None, {"image": img[None,]})[0]
|
||||
bm = DOC_TR.run(None, {'image': img[None,]})[0]
|
||||
bm = paddle.to_tensor(bm)
|
||||
bm = paddle.nn.functional.interpolate(
|
||||
bm, y.shape[2:], mode="bilinear", align_corners=False
|
||||
bm, y.shape[2:], mode='bilinear', align_corners=False
|
||||
)
|
||||
bm_nhwc = np.transpose(bm, (0, 2, 3, 1))
|
||||
out = paddle.nn.functional.grid_sample(y, (bm_nhwc / 288 - 0.5) * 2)
|
||||
Reference in New Issue
Block a user