PyTorch
YF-Zhang
这个作者很懒,什么都没留下…
展开
-
PyTorch Learning (一)
PyTorch Learning本人有一些python基础,听说过pytorch但没用过。下面记录一下pytorch学习,方便后面自己查阅。一些相关链接PyTorch中文文档: https://pytorch-cn.readthedocs.io/zh/latest/PyTorch入门教程-简书:https://www.jianshu.com/u/5e2b32ff790c回到正题一 安装...原创 2019-01-06 09:29:23 · 289 阅读 · 0 评论 -
学习Faster R-CNN代码faster_rcnn(八)
Faster R-CNN源代码中faster_rcnn文件夹中包含三个文件 faster_rcnn.py,resnet.py,vgg16.py。【占坑,未完待续…】原创 2019-02-26 10:57:06 · 1239 阅读 · 0 评论 -
学习Faster R-CNN代码nms(七)
非极大值抑制(NMS)原创 2019-02-25 23:09:33 · 1304 阅读 · 0 评论 -
学习Faster R-CNN代码rpn(六)
RPN是Faster R-CNN的一个重点。本人前面的博客《 Faster R-CNN》( https://blog.csdn.net/weixin_43872578/article/details/85869629 )提到RPN的实现。原创 2019-02-23 22:22:58 · 1755 阅读 · 3 评论 -
学习Faster R-CNN代码roi_align(五)
ROI_ALIGN?roi align是对roi pooling的改进,roi pooling是最邻近插值,在映射时如果是小数则简单省略了小数部分。而roi align是采用了双线性插值,使得即使缩放后坐标不能刚好为整数,也能通过插值得到浮点数处的值处理得到pooling后的值。详细说明roi pooling和roi align可参考‘AI深度学习求索’ https://baijiahao.b...原创 2019-02-23 22:22:25 · 1414 阅读 · 0 评论 -
学习Faster R-CNN代码roi_crop(四)
为什么要crop由于传统的CNN限制了输入必须固定大小(比如AlexNet是224x224),所以在实际使用中往往需要对原图片进行crop或者warp的操作:crop:截取原图片的一个固定大小的patchwarp:将原图片的ROI缩放到一个固定大小的patch无论是crop还是warp,都无法保证在不失真的情况下将图片传入到CNN当中:crop:物体可能会产生截断,尤其是长宽比大的...原创 2019-02-23 22:21:56 · 1406 阅读 · 0 评论 -
学习Faster R-CNN代码demo(一)
注释Yang Jianwei 大神的Faster R-CNN代码先放github: https://github.com/jwyang/faster-rcnn.pytorch原创 2019-01-23 11:55:48 · 2133 阅读 · 5 评论 -
Pytorch搭建网络
用PyTorch搭建一个小网络下面以计算两个数的和为例,x是输入数据,y是标签。import torchimport torch.nn as nnx = torch.Tensor([[0.2,0.4],[0.2,0.3],[0.3,0.4]])#数据y = torch.Tensor([[0.6],[0.5],[0.7]])#标签 不需要求导#定义网络class MyNet(nn....原创 2019-01-21 23:39:32 · 210 阅读 · 0 评论 -
几种网络LeNet、VGG Net、ResNet原理及PyTorch实现
LeNetLeNet比较经典的一张图如下图LeNet-5共有7层,不包含输入,每层都包含可训练参数;每个层有多个Feature Map,每个FeatureMap通过一种卷积滤波器提取输入的一种特征,然后每个FeatureMap有多个神经元。INPUT层-输入层输入图像的尺寸统一归一化为: 32 x 32。C1层 卷积层输入图片:32 x 32卷积核大小:5 x 5卷积核种类:6...原创 2019-01-20 16:24:02 · 2564 阅读 · 0 评论 -
学习Faster R-CNN代码roi_pooling(三)
这一篇单独拿出来了解一下roi_pooling/src/roi_pooling.c中C代码:** ## roi_pooling/src/roi_pooling.c ## **#include <TH/TH.h>#include <math.h&am原创 2019-01-25 00:31:50 · 1128 阅读 · 1 评论 -
学习Faster R-CNN代码roi_pooling(二)
roi_pooling目录下-src文件夹下是c和cuda版本的源码,其中roi_pooling的操作的foward是c和cuda版本都有的,而backward仅写了cuda版本的代码。-functions文件夹下的roi_pool.py是继承了torch.autograd.Function类,实现RoI层的foward和backward函数。-modules文件夹下的roi_pool....原创 2019-01-23 20:08:17 · 974 阅读 · 0 评论 -
PyTorch Learning (二)
自动求导机制(autograd)原创 2019-01-20 23:39:48 · 165 阅读 · 0 评论 -
复现 faster rcnn
已经好几天没接着看faster rcnn的源代码了,前几天试了一下可以成功训练了。jwyang’s github: https://github.com/jwyang/faster-rcnn.pytorch记录一下怎么做的吧。凭印象回忆一下吧。虚拟环境安装虚拟环境 pip install --user virtualenv进入虚拟环境 source zyfenv/bin/activate...原创 2019-04-18 15:53:37 · 545 阅读 · 1 评论