![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
负壹
这个作者很懒,什么都没留下…
展开
-
YOLO模型详解
简介 YOLO模型是用来目标检测的,也就是在一副图像中检测到物体并识别出物体在图像中所在的位置,YOLO相比于以往的带有建议框的神经网络,速度大有提升,下边详细介绍YOLO网络YOLO网络层 网络还是比较简单的,从448*448*3的图像依次经过途中所有卷积层与池化层得出了7*7*1024的特征图,再经过两层全连接层得到7*7*30的输出结果 下边解释一下为什么输出层数是30,当我们得到7*7*1024的特征图时,对每一个网格都做一次预测,也就...原创 2020-08-09 16:02:57 · 18977 阅读 · 1 评论 -
Pycharm运行中忽略warning
在代码中加入import warningswarnings.filterwarnings("ignore")就可以避免warning弹出了原创 2020-08-09 09:18:41 · 8307 阅读 · 0 评论 -
Batch Size的理解
1、为什么用batchsize 当网络在训练的时候,如果将样本一个一个输入,网络对每一个输入都会计算一个梯度,不同样本梯度不同,会使得梯度方向不准确甚至跳动,所以需要同时输入网络batchsize大小个样本,这样计算的梯度是由输入样本一起决定的,有一种类似平均的概念在里边,可以避免极端数据造成的梯度误差2、batchsize影响 batchsize不能太大,因为GPU很难装下那么多数据,如果batchsize太小,可能会造成梯度方向不准确;实际上,当batchsize...原创 2020-08-08 20:11:17 · 1606 阅读 · 0 评论 -
[解决方案] pytorch中RuntimeError: DataLoader worker (pid(s) 27292) exited unexpectedly
在运行代码前边加上:if __name__ == '__main__':就可以了原创 2020-08-08 20:03:52 · 17220 阅读 · 17 评论 -
argparse常用用法讲解
前言 在看开源的python代码的时候,经常发现命令行输入和解析的例子,比如我当前调用的pointnet的源码,也是这样,平时总是手动将参数输入在编译器里,但是这次我准备把这个命令行输入好好学习一下 argparseargparse是专门用于从命令行中解析参数的一个库,举一个例子,如下代码:(文件名为:exam_argparse.py)import argparseparser = argparse.ArgumentParser()parse...原创 2020-07-17 16:57:57 · 1413 阅读 · 1 评论 -
[论文解读+复现] PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
一、摘要 本文提出了一种结构简单的物体检测与点云分割、语义分析的深度学习网络,网络具有置换不变性,文章最后对网络的鲁棒性做出了解释。 二、介绍 典型卷积网络需要规则的输入,例如像素和体素,目的是可以共享权重参数(weight),一些研究学者会将点云转化为体素或者图像集再传入神经网络,这会导致数据量变得巨大并造成模糊;为此本文使用简单的统一的结构—点云来进行训练,网络名为:PointNe...原创 2020-07-11 15:40:06 · 6508 阅读 · 3 评论 -
PointNet模型的Pytorch代码详解
前言 关于PointNet模型的构成、原理、效果等等论文部分内容,我在之前一篇论文中写到过,可以参考这个链接:PointNet论文笔记 下边我就直接放一张网络组成图,并对代码进行解释,我以一种比较容易理解的顺序放置,希望耐心阅读。网络结构图示 在分类网络中,输入n个点,对输入做特征变换,再进行最大池化输出k个种类;分割网络是分类网络的一个拓展,它考虑了全局和局部的特征以及每个点的输出分数。mlp代表多层感知机,括号中是感知机的层数,批标准化(Batch...原创 2020-07-17 15:29:00 · 10519 阅读 · 17 评论 -
FCN模型实现-Pytorch+预训练VGG16
FCN模型的网络与VGG16类似,之后后边将全连接层换成了卷基层,具体的网络结构与细节可以去看论文:https://people.eecs.berkeley.edu/~jonlong/long_shelhamer_fcn.pdf下边详细讲一下用Pytorch对FCN的实现:本文参考了https://zhuanlan.zhihu.com/p/32506912但是修改了部分代码,加...原创 2020-03-17 14:57:53 · 6281 阅读 · 7 评论 -
Kaggle数据集猫狗分类(Pytorch+ResNet34预训练)99%以上正确率
关于训练集的介绍和数据划分可以参照上一个博客:https://blog.csdn.net/qq_41685265/article/details/104895273数据加载class DogCat(data.Dataset): def __init__(self, root, transforms=None, train=True, test=False): ...原创 2020-03-16 16:06:37 · 9953 阅读 · 13 评论 -
狗猫分类数据集划分详解
数据集介绍首先是要下载数据集,下载地址:数据解压之后会有两个文件夹,一个是“train”,一个是“test”,顾名思义一个是用来训练的,另一个是作为检验正确性的数据,也是网站要求提交标签的。在train文件夹里边是一些已经命名好的图像,有猫也有狗而在test文件夹中是只有编号名的图像大致了解了数据集后,下边就开始划分数据集代码先放一段代码,这是从书中截取出来...原创 2020-03-16 11:44:34 · 9765 阅读 · 4 评论 -
Pytorch实现鲜花分类(102 Category Flower Dataset)
本文主要讲解该算法的实现过程实验环境python3.6+pytorch1.2+cuda10.1数据集102 Category Flower Dataset数据集由102类产自英国的花卉组成,每类由40-258张图片组成下边是代码实现过程及讲解数据加载#选择设备device = torch.device("cuda:0")#对三种数据集进行不同预处理,对训练数据进行...原创 2020-03-09 09:40:19 · 10917 阅读 · 34 评论 -
Pytorch实现图像风格迁移
载入图像imsize = 512loader = transforms.Compose([ transforms.Resize(imsize), transforms.ToTensor()])def image_loader(image_name): image = Image.open(image_name) image = loader(image)...原创 2020-03-08 10:09:18 · 2388 阅读 · 3 评论 -
Pytorch用LeNet在CIFAR10数据集上实现分类
本文主要讲解该算法的实现过程实验环境python3.6+pytorch1.2+cuda10.1数据集CIFAR10数据集包含6万训练图像与1万验证图像下边是代码实现过程及讲解数据加载#载入数据def loadCIFAR10(batch_size): transform = transforms.Compose( [transforms...原创 2020-03-07 20:50:07 · 737 阅读 · 0 评论 -
Pytorch用LeNet在MNIST数据集上实现手写数字识别
本文主要讲解该算法的实现过程实验环境python3.6+pytorch1.2+cuda10.1数据集MNIST数据集包含6万训练图像与1万验证图像下边是代码实现过程及讲解数据加载#载入数据def loadMNIST(batch_size): trans_img = transforms.Compose([ transforms.ToTe...原创 2020-03-07 16:39:33 · 2185 阅读 · 0 评论 -
Pytorch使用GPU计算
方法一可以使用.cuda()将数据转换到GPU中x = torch.zeros(3,3,dtype=torch.double)y = torch.ones(3,3,dtype=torch.double)if torch.cuda.is_available(): x = x.cuda() y = y.cuda() z = x+y print(z)#将...原创 2020-03-07 13:47:16 · 3908 阅读 · 1 评论 -
Torch的Tensor与numpy的array数据相互转换
Torch的Tensor与numpy的array会共享储存空间,修改一个也会导致另外一个被修改Tensor转换成Numpy数组a = torch.ones(5)b = a.numpy()a.add_(1)print(a)print(b)将Numpy数组转换成Tensora = np.ones(5)b = torch.from_numpy(a)np...原创 2020-03-07 13:37:45 · 5397 阅读 · 0 评论 -
pytorch快速入门--构建线性回归
准备数据 这里用一组比较简单的数据x_data = torch.Tensor([[1.0],[2.0],[3.0]])y_data = torch.Tensor([[3.0],[6.0],[9.0]])构建线性回归模型#定义一个线性回归类class Model(torch.nn.Module): def __init__(self): ...原创 2020-03-07 13:26:10 · 347 阅读 · 0 评论 -
英伟达(NVIDIA)显卡、驱动版本与cuda版本对应关系
首先给出驱动版本与cuda版本的对应关系:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html查看自己的显卡驱动版本使用命令nvidia-smi如果想要查看自己的显卡能安装的最新驱动,去这个网站查找:https://www.geforce.cn/drivers...原创 2020-03-05 10:27:31 · 19765 阅读 · 0 评论 -
conda在指定环境下安装包
有两个方法可以实现:方法一:conda install -n 环境名 包名举个例子,在名字为test_env的环境中安装tensorflowxxxx$ conda install -n test_env tensorflow方法二:进入环境,在环境里用conda安装举个例子,在名字为test_env的环境中安装tensorflow先进名字为test_env的...原创 2020-03-05 18:35:45 · 6750 阅读 · 2 评论