pytorch
骚年,你渴望力量嘛?
在读研究生,初来乍到,多多指教。
展开
-
pytorch入门之tensorboard
主要使用两个工具,一个是add_scalar图表,另一个是add_graph网络。主要是基于pytorch入门之CIFAR-10图像分类更改而来原创 2020-11-20 09:40:16 · 319 阅读 · 0 评论 -
pytorch入门之CIFAR-10图像分类
目录 CIFAR-10: 实现思路: 加载数据集: 定义网络: CIFAR-10: 一共包含 10 个类别的 RGB 彩色图 片:飞机( a叩lane )、汽车( automobile )、鸟类( bird )、猫( cat )、鹿( deer )、狗( dog )、蛙类( frog )、马( horse )、船( ship )和卡车( truck )。图片的尺寸为 32×32 ,数据集中一共有 50000 张训练圄片和 10000 张测试图片。 实现思路: 使用torchvision.原创 2020-10-31 21:14:31 · 2470 阅读 · 4 评论 -
pytorch入门之MNIST数据集分类
本着入门pytorch的热情,实现pytorch的"helloword" 概述:搭建LeNet-5卷积神经网络,该网络具体参数配置见代码。 代码详解,代码有详细注释: ps:我这是用GPU跑的代码,如果要用cpu跑,则需要把train中的.cuda()一行删除,并适当降低卷积神经元的个数。 mycnn部分: import torch.nn as nn class MyCNN(nn.Module): # 继承module def __init__(self): # 定义构造函数原创 2020-10-04 21:19:36 · 426 阅读 · 0 评论 -
pytorch入门之tensor基础操作
创建tensor 创建指定元素的tensor: torch.tensor(data, dtype=None, device=None, requires_grad=False, pin_memory=False) 创建随机元素的tensor: torch.rand(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) 创建服从正态分布N(0,1)的tensor: torch.randn(.原创 2020-10-03 21:56:18 · 1520 阅读 · 0 评论 -
fast-rcnn 源码解读(pytorch版)
Pytorch源码:https://github.com/chenyuntc/simple-faster-rcnn-pytorch 第一位作者的解读: (一)读取和数据处理 https://blog.csdn.net/qq_32678471/article/details/84776144 (二)模型准备 https://blog.csdn.net/qq_32678471/article/details/84792307 (三)模型创建 https://blog.csdn.net/qq_32678471/a转载 2020-09-30 19:38:22 · 1857 阅读 · 0 评论