flask部署pytorch-客户端

举报
Nikolas 发表于 2021/01/03 21:33:32 2021/01/03
【摘要】 用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

```

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。