文章目录

1. 需要用到的库
2. 模型定义
3. 测试

基于上一篇文章 https://blog.csdn.net/linghu8812/article/details/100044971，这次介绍一下网络模型的定义。

1. 需要用到的库

import torch
import torch.nn as nn
from torchvision.models import resnet50

基于ResNet50定义网络模型，在此基础上进行微调。

2. 模型定义

class CatVSDogNet(nn.Module):def __init__(self, num_class=2):super(CatVSDogNet, self).__init__()self.model = resnet50(pretrained=True)self.model.fc = nn.Linear(2048, num_class)def forward(self, x):output = self.model(x)return output

基于ImageNet进行初始化，ImageNet有1000类，将网络模型的全连接层修改为两类。

3. 测试

测试代码如下：

def main():net = CatVSDogNet()y = net(torch.randn(5, 3, 448, 448))print(y.size())if __name__ == '__main__':main()

执行代码后输出：

torch.Size([5, 2])

基于PyTorch的卷积神经网络图像分类——猫狗大战（二）：使用Pytorch定义网络模型相关推荐

【深度学习】基于Pytorch的卷积神经网络概念解析和API妙用(二)
[深度学习]基于Pytorch的卷积神经网络API妙用(二) 文章目录1 Padding和Stride 2 多输入多输出Channel 3 1*1 Conv(笔者在看教程时,理解为降维和升维) 4 池 ...
【深度学习】基于Pytorch的卷积神经网络概念解析和API妙用(一)
[深度学习]基于Pytorch的卷积神经网络API妙用(一) 文章目录 1 不变性 2 卷积的数学分析 3 通道 4 互相关运算 5 图像中目标的边缘检测 6 基于Pytorch的卷积核 7 特征映射 ...
用PyTorch搭建卷积神经网络
用PyTorch搭建卷积神经网络本篇是加拿大McGill本科,Waterloo硕士林羿实习时所作的工作.发文共享,主要是面对PyTorch的初学者. 本篇文章是一篇基础向的PyTorch教程,适合有 ...
卷积神经网络：（二）风格迁移——原理部分
卷积神经网络:(二)风格迁移--原理部分引言本文是在第一步配置完环境后基础上运行的.使用的为系统直装的python环境(在anaconda环境下一样适用,后面注意的点会提示的.).若想查看环境配置 ...
【PyTorch】卷积神经网络
卷积神经网络卷积神经网络最初是为了解决计算机视觉的相关问题设计的,现在其不仅被应用在图像和视频领域,也可用于诸如音频信号的时间序列信号的处理之中. 本文主要着重于卷积神经网络的基本原理和使用PyTo ...
Pytorch实现卷积神经网络（回归问题）
Pytorch实现卷积神经网络(回归问题) 前言一.Pytorch 二.卷积神经网络 1.CNN结构 2.实现一个卷积神经网络类--代码总结前言深度学习当前的热度虽然没有前两年那么火热,尤其是 ...
卷积神经网络新手指南之二
导语:本文将进一步探讨有关卷积神经网络的更多细节. 卷积神经网络新手指南之二引言本文将进一步探讨有关卷积神经网络的更多细节,注:以下文章中部分内容较为复杂,为了保证其简明性,部分内容详细解释的研究 ...
基于Keras的卷积神经网络（CNN）可视化
基于Keras的卷积神经网络(CNN)可视化标签(空格分隔): 深度学习卷积神经网络可视化本文整理自Deep Learning with Python,书本上完整的代码在这里的5.4节,并陪有 ...
基于tensorflow和卷积神经网络的电影推荐系统的实现
基于tensorflow和卷积神经网络的电影推荐系统的实现一. 数据处理 1. MovieLens数据集 2. 处理后的数据二. 建模&训练 1. 嵌入层 2. 文本卷积层 3. 全连接层 ...

基于PyTorch的卷积神经网络图像分类——猫狗大战（二）：使用Pytorch定义网络模型

文章目录

1. 需要用到的库

2. 模型定义

3. 测试

基于PyTorch的卷积神经网络图像分类——猫狗大战（二）：使用Pytorch定义网络模型相关推荐

最新文章

热门文章