[python]将图像处理程序改写成webapi

遗憾迟香

python菜鸟,在github上发现了一个很厉害的人工智能图片处理工具图像处理部分源代码:

[Python] 纯文本查看 复制代码

def run(input_img):
    input_img = cv2.cvtColor(input_img, cv2.COLOR_BGR2RGB)
    out_img = np.array(input_img, dtype=np.uint8)
    face = f.face_align(out_img)
    if len(face) > 0:
        print('face detected')
        orig_face_img = face["image"]
        H, W, _ = orig_face_img.shape
        face_img = cv2.resize(orig_face_img, (512, 512))
        face_img = generate(face_img, mask_img, 0.7)
        face_img = cv2.resize(face_img, (W, H))
        out_img = f.face_inverse(face_img, out_img, face["matrix"])
    else:
        print('face not detected')
    out_img = cv2.cvtColor(out_img, cv2.COLOR_BGR2RGB)
    return out_img

import os

print('hello world')

if not os.path.exists('output_photos'):
    os.mkdir('output_photos')

for pic in os.listdir('photos'):#for each picture in the 'photos' folder
    input_pic = cv2.imread('photos' + "/" + pic, 1) # loads the picture
    if input_pic is None:
        continue
    output_pic = run(input_pic)
    output_pic_name = 'output_photos/'+pic
    cv2.imwrite(output_pic_name,output_pic)

该段代码可以实现批量处理photos目录下的图片,输出到output_photos
我想把它改写成网页版,完成服务端即可
想要实现的功能为
客户端使用post上传图片,服务端接收到图片之后进行处理,然后将处理过的图片链接返回给客户端
使用flask实现

清风浮云net

直接封成wasm应该可以直接在网页端调用

E式丶男孩

那你用flask做个web接口不就好了。你都知道怎么做了

rhci

很牛，但是不知道怎么用，先看看大佬怎么写吧

yjn866y

嗯。进来学习的