计算机视觉-分类网络的构建
学习用pytorch构建分类网络
呆呆珝
(1) 加群交流:295248651
(2)可以兼职,欢迎打扰,白嫖绕道
展开
-
基础概念:图片的卷积可视化结果
卷积操作可视化原创 2023-07-31 11:13:52 · 203 阅读 · 0 评论 -
3.2分类训练之AlexNet网络构建(Pytorch)
前言 之前的文章介绍过AlexNet的网络结构,今天通过pytorch框架实现一下跳连结构的Resnet18网络结构。具体的代码是下面这个链接博主写的,我是进行注释一下,代码不是原创。原创 2021-06-17 10:05:20 · 209 阅读 · 0 评论 -
3.1分类训练之Resnet网络构建(Pytorch)
前言 之前的文章介绍过VGG16的网络结构,今天通过pytorch框架实现一下串联的VGGNET网络结构。原创 2021-06-15 17:57:36 · 815 阅读 · 4 评论 -
4.分类训练之训练脚本(Pytorch)
import torchimport torch.nn as nnimport torchvisionfrom vggnet import VGGNetfrom load_cifar10 import train_loader, test_loaderimport os# 是否使用GPUdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")# 遍历样本200次epoch_num =...原创 2021-06-10 17:48:06 · 1168 阅读 · 2 评论 -
3.分类训练之VGGNET网络构建(Pytorch)
import torch# 进行卷积import torch.nn as nn# 进行softmaximport torch.nn.functional as Ffrom torchsummary import summaryclass VGGbase(nn.Module): # 初始化函数 def __init__(self): # 初始化类 super(VGGbase, self).__init__() # 定义算子 .原创 2021-06-10 14:59:52 · 534 阅读 · 0 评论 -
2.分类训练之数据准备-自定义数据集(CIFAR10数据加载)
前言 pytorch中有许多公开的数据集可以加载,但是换成自己数据集,想训练自己定义的类别怎样做?这里以CIFAR10解析出来的图片数据为例。原创 2021-06-10 13:50:41 · 1536 阅读 · 0 评论 -
1.分类训练之数据准备-CIFAR10数据解析
CIFAR10数据介绍 该数据集共有60000张彩色图像,这些图像是32*32,分为10个类,每类6000张图。这里面有50000张用于训练,构成了5个训练批,每一批10000张图;另外10000用于测试,单独构成一批。测试批的数据里,取自10类中的每一类,每一类随机取1000张。抽剩下的就随机排列组成了训练批。注意一个训练批中的各类图像并不一定数量相同,总的来看训练批,每一类都有5000张图。...原创 2021-06-10 11:30:57 · 558 阅读 · 3 评论