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

写在前面

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

成果演示

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

接口文档

请求地址

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

请求方法

post

必要参数

成功响应

请求示例

测试图片

请求代码

import requests, base64, json

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))
1
2
3
4
5
6
7
8
9

响应结果

{
	"prediction": 2,
    "confidence": "98.30%"
}
1
2
3
4

温馨提示

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