修改ResNet-18网络(使用预训练),输入通道为1,输出为2,并返回前连接层的特征数和最后的输出:
import torch
import torch.nn as nn
import torchvision.models as models# 定义自定义的ResNet-18网络
class…
数据集下载
完整数据集共25000张图片Download Kaggle Cats and Dogs Dataset from Official Microsoft Download Center
数据划分
数据集结构 导包
import os
import torch
import torch.nn as nn
import torch.nn.parallel
import torchvision.models
import torch.optim …