【python】手写数字识别模型api接口调用

写在前面

我用pytorch搭建了一个cnn模型，训练mnist数据集，然后将训练后的模型部署到了线上，并提供一个可供调用的api接口。

成果演示

http://pytorch-cnn-mnist.herokuapp.com/

接口文档

• 请求地址

http://pytorch-cnn-mnist.herokuapp.com/predict/

• 请求方法

post

• 请求参数

字段	类型	描述
data	字符串	图片的base64编码（包含头部信息）

• 成功响应

字段	类型	描述
prediction	整型	手写数字的预测结果
confidence	字符串	预测结果正确的概率

请求示例

• 测试图片

• 请求代码

import requests
import base64
import json

# test.jpg就是测试图片
with open('test.jpg', 'rb') as f:
    data = base64.b64encode(f.read()).decode()

# 加上头部信息
data = 'data:image/jpeg;base64,'+data
url = 'http://pytorch-cnn-mnist.herokuapp.com/predict/'
res = requests.post(url, data=data).json()
print(json.dumps(res, indent=4))

• 响应结果

{
	"prediction": 2,
    "confidence": "98.30%"
}

温馨提示

接口仅供调用测试，有次数限制，非法滥用会封IP，请勿用于实际生产环境。