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

```