python利用pytorch实现图像识别分类，并搭建简单的服务器

pip install pytorch

2、代码如下

import torch
import torchvision
from PIL import Image
from torch import nn
from torchvision import transforms
import os
from flask import Flask, request
app = Flask(__name__)
from flask import jsonify
from werkzeug.utils import secure_filename
# 上传的图片保存路径
UPLOAD_PATH = os.path.join(os.path.dirname(__file__), 'images')

normalize = transforms.Normalize(mean=[.5, .5, .5], std=[.5, .5, .5])  # 规范化
transforms = transforms.Compose([transforms.Resize((64, 64)),
                                     torchvision