pytorch的flask部署python调用

官方github：https://github.com/avinassh/pytorch-flask-api
服务器app.py代码：

import io
import json

from torchvision import models
import torchvision.transforms as transforms
from PIL import Image
from flask import Flask, jsonify, request

"""
Run the Flask server:
FLASK_ENV=development FLASK_APP=app.py flask run
From another tab, send the image file in a request:
curl -X POST -F file=@cat_pic.jpeg http://localhost:5000/predict
"""

app = Flask(__name__)
imagenet_class_index = json.load(open('imagenet_class_index.json'))
model = models.densenet121(pretrained=True)
model.eval()


def transform_image(image_bytes):
    my_transforms = transforms.Compose([transforms.Resize(255),
                                        transforms.CenterCrop(224),
                                        transforms.ToTensor(),
                                        transforms.Normalize(
                                            [0.485, 0.456, 0.406],
                                            [0.229, 0.224, 0.225])])
    image = Image.open(io.BytesIO(image_bytes))
    return my_transforms(image).unsqueeze(0)


def get_prediction(image_bytes):
    tensor = transform_image(image_bytes=image_bytes)
    outputs = model.forward(tensor)
    _, y_hat = outputs.max(1)
    predicted_idx = str(y_hat.item())
    return imagenet_class_index[predicted_idx]


@app.route('/predict', methods=['POST'])
def predict():
    if request.method == 'POST':
        file = request.files['file']
        img_bytes = file.read()
        class_id, class_name = get_prediction(image_bytes=img_bytes)
        return jsonify({'class_id': class_id, 'class_name': class_name})


if __name__ == '__main__':
    app.run()```

客户端调用client.py

```python
import requests
import os

# image = open(r"E:\pycharm_project\Data-proessing\model_flask\timg.jpg", 'rb').read()
# payload = {'file': image}
#
# r = requests.post("http://localhost:5000/predict", files=payload).json()
# print(r)

for i in os.listdir("test_img"):
    image = open("test_img/"+i,'rb')
    payload = {'file':image}
    r = requests.post("http://localhost:5000/predict", files=payload).json()
    print(r)

git bash控制台：
启动flask服务器：FLASK_ENV=development FLASK_APP=app.py flask run
测试命令：curl -X POST -F file=@test_img/dog.jpg http://localhost:5000/predict