图像分类
令狐傻笑
图像算法工程师,主要研究方向为图像分类,目标检测
展开
-
基于PyTorch的卷积神经网络图像分类——猫狗大战(二):使用Pytorch定义网络模型
文章目录1. 需要用到的库2. 模型定义3. 测试基于上一篇文章https://blog.csdn.net/linghu8812/article/details/100044971,这次介绍一下网络模型的定义。1. 需要用到的库import torchimport torch.nn as nnfrom torchvision.models import resnet50基于ResNet50定义网络模型,在此基础上进行微调。2. 模型定义class CatVSDogNet(nn.Module原创 2021-07-27 17:08:19 · 785 阅读 · 0 评论 -
基于PyTorch实现cosine learning rate
1.需要用到的库设置学习率和模型import mathimport matplotlib.pyplot as pltimport torch.optim as optimfrom torchvision.models import resnet18lr_rate = 0.1model = resnet18(num_classes=10)2.LambdaLR实现cosine...原创 2020-03-26 16:49:55 · 6401 阅读 · 0 评论