26 lines
614 B
Python
26 lines
614 B
Python
import os
|
|
|
|
import cv2
|
|
from flask import Flask, request
|
|
|
|
from paddle_services.doc_dewarp import dewarper
|
|
from utils import process_request, parse_img_path
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route('/dewarp', methods=['POST'])
|
|
@process_request
|
|
def dewarp():
|
|
img_path = request.form.get('img_path')
|
|
img = cv2.imread(img_path)
|
|
dewarped_img = dewarper.dewarp_image(img)
|
|
dirname, img_name, ext = parse_img_path(img_path)
|
|
save_path = os.path.join(dirname, img_name + '_dewarped.' + ext)
|
|
cv2.imwrite(save_path, dewarped_img)
|
|
return save_path
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app.run('0.0.0.0', 5007)
|