[python]将图像处理程序改写成webapi
本帖最后由 遗憾迟香 于 2023-1-31 12:04 编辑python菜鸟,在github上发现了一个很厉害的人工智能图片处理工具图像处理部分源代码:
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实现 直接封成wasm应该可以直接在网页端调用 那你用flask做个web接口不就好了。你都知道怎么做了 本帖最后由 rhci 于 2023-1-31 15:26 编辑
很牛，但是不知道怎么用，先看看大佬怎么写吧 嗯。进来学习的
页:
[1]