faster-rcnn
文章平均质量分 57
冰菓(笑)
希望写一些有质量的东西
展开
-
pytorch 从头开始faster-rcnn(五):roi
更新:2019/8/12 上文说到 最终获得的检测框将用于ROI层真值标签的生成,以及进行筛选3:1的正负样本放入ROI层进行训练(共128个); 同时进入ROI层的还有vgg16生成的特征图,这个特征图用于上面说的3:1正负样本坐标将相应位置的特征图抠出来用于ROIpooling层进行金字塔池化,通过金字塔池化生成固定尺寸大小(7*7)特征图;在用来全连接层分类和定位,计算损失方式与PRN...原创 2018-10-02 21:12:55 · 1937 阅读 · 0 评论 -
pytorch 从头开始faster-rcnn(一):代码知识准备
本文代码是参考github上的simple-faster-rcnn-pytorch:https://github.com/chenyuntc/simple-faster-rcnn-pytorch 有需要的可以自行下载参考,就不放上自己写的,上面的已经写的很详细了!!! 一: class config: def _parse(self, kwargs): state_...原创 2018-09-18 12:47:51 · 5459 阅读 · 3 评论 -
pytorch 从头开始faster-rcnn(二):数据集处理
一个神经网络的开始首先是要对数据进行处理。 目标检测图像分为两个部分,一个是原始图像,另外一个是标志的xml文件。 由于pytorch数据使用迭代器加载,所以处理都是一张一张图像的处理。 1.加载图像和标签。 标签包括boundingbox和其名称标签,由于boundingbox和其标签有多个,所有使用循环读取。返回图像及其标签。 def get_example(self, i...原创 2018-09-20 14:11:55 · 2220 阅读 · 2 评论 -
pytorch 从头开始faster-rcnn(零):使用到的函数
本篇博客是在写faster-rcnn遇到的没见过的函数,所以这篇博客随着代码的编写不定期更新。 1.tqdm tqdm在阿拉伯语中的意思是进展。tqdm可以在长循环中添加一个进度提示信息,用户只需要封装任意的迭代器 tqdm(iterator),是一个快速、扩展性强的进度条工具库。 from tqdm import tqdm from time import sleep for i in...原创 2018-09-20 15:09:39 · 1180 阅读 · 0 评论 -
pytorch 从头开始faster-rcnn(三):vgg16 (带有网络冻结的写法)
由于pytorch自带了一些网络结构可以直接使用,所以直接调用有的模型,详细的可以查看官方文档:https://pytorch.org/docs/stable/torchvision/models.html def decom_vgg16(): if opt.caffe_pretrain: model = vgg16(pretrained=False) ...原创 2018-09-26 15:23:00 · 5437 阅读 · 0 评论 -
pytorch 从头开始faster-rcnn(四):rpn
具体参数可以查看:https://www.cnblogs.com/wangyong/p/8513563.html RPN流程: 1.每一张图片生成固定数量的锚节点,锚节点个数是最后一次特征图大小,比如说最后一层特征大小为(37,50),所以有37*50个锚节点.锚节点之间的步长为16像素点,这是因为vgg16有4次maxpool,所有图像缩小了16倍,所以步长为16像素。每一个锚节点又生成9个...原创 2018-09-26 19:29:54 · 3490 阅读 · 0 评论 -
pytorch 从头开始faster-rcnn 最后:使用自己的数据集
本文代码是参考github上的simple-faster-rcnn-pytorch:https://github.com/zylhub/simple-faster-rcnn-pytorch 由于代码原作者能力太强 ,我太弱了,很多看不懂,所有先提前把使用放上来供大家参考。 修改2个地方就可以使用自己的数据集,不过数据集需要制作成voc2007里面的格式。 1.将voc_dataset.py和...原创 2018-10-02 21:23:35 · 2562 阅读 · 23 评论