ResNet 图片分类并且部署成Web

目录

1. 介绍

2. model 部分

3. predict 部分

4. main 部分

5. 本地Web

6. 部署到Web

---

1. 介绍

ResNet 网络的搭建可以看之前的文章：ResNet 训练CIFAR10数据集，并做图片分类

github 的代码在这里：My_github

制作的分类网站：ResNet 图像分类

2. model 部分

ResNet 网络的代码为：

import torch.nn as nn
from torch.nn import functional as F


class ResidualBlock(nn.Module):     # 搭建 残差结构
    def __init__(self, inchannel, outchannel, stride=1, shortcut=None):
        super(ResidualBlock, self).__init__()
        self.left = nn.Sequential(
            # stride