flask部署pytorch-客户端
## 1. 导入依赖包
```python
import requests
import argparse
```
## 2. 定义请求URL
这里用本机URL做例子,实际使用中要改成服务器的IP地址
```python
PyTorch_REST_API_URL = 'http://127.0.0.1:5000/predict'
```
## 3. 发送图片,获取返回结果
首先读取一张本地图片,然后发送给服务器
```python
def predict_result(image_path):
image = open(image_path, 'rb').read()
payload = {'image': image}
r = requests.post(PyTorch_REST_API_URL, files=payload).json()
if r['success']:
for (i, result) in enumerate(r['predictions']):
print('{}. {}: {:.4f}'.format(i + 1, result['label'], result['probability']))
else:
print('Request failed')
```
## 4. 定义主函数
```python
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='demo')
parser.add_argument('--file', type=str, help='test') # 获取终端file=后面的字符串
args = parser.parse_args()
predict_result(args.file)
```
运行本文件,将要测试的图片路径传入
### python 本文件名.py --file=./data/dog.jpg #路径不加引号
```python
```
- 点赞
- 收藏
- 关注作者
评论(0)