原文:r-cnn学习(四):train_faster_rcnn_alt_opt.py源码学习

论文看的云里雾里,希望通过阅读其代码来进一步了解。 参考:http: blog.csdn.net sloanqin article details 首先是. tools train faster rcnn alt opt.py,通过其main函数了解整个训练流程。 通过上面的代码可以看出,整个迭代过程分为四步 参考论文 。其中后面两步固定共享卷积 层,只对RPN和fc层进行微调。 接着看看每一步是 ...

2016-12-04 16:00 0 2807 推荐指数:

查看详情

学习Faster R-CNN代码faster_rcnn(八)

Faster R-CNN源代码中faster_rcnn文件夹中包含三个文件 faster_rcnn.py,resnet.py,vgg16.py。 1.faster_rcnn.py注释 ref:https://blog.csdn.net/weixin_43872578 ...

Fri Aug 16 03:33:00 CST 2019 0 371
学习Faster R-CNN代码rpn(六)

代码文件结构 bbox_transform.py # bounding box变换。 generate_anchors.py # 生成anchor,根据几种尺度和比例生成的anchor。 proposal_layer.py # 通过将估计的边界框变换应用于一组常规框 ...

Fri Aug 16 03:12:00 CST 2019 0 908
学习Faster R-CNN代码nms(七)

非极大值抑制(Non-Maximum Suppression NMS) NMS就是去除冗余的检测框,保留最好的一个。 产生proposal后使用分类网络给出每个框的每类置信度,使 ...

Fri Aug 16 03:16:00 CST 2019 0 562
学习Faster R-CNN代码demo(一)

注释Yang Jianwei 的Faster R-CNN代码(PyTorch) jwyang’s github: https://github.com/jwyang/faster-rcnn.pytorch 文件demo.py 这个文件是自己下载好训练好的模型后可执行 下面是对代码的详细 ...

Thu Aug 15 01:31:00 CST 2019 0 1138
学习Faster R-CNN代码roi_pooling(三)

这一篇单独拿出来了解一下roi_pooling/src/roi_pooling.c中C代码: 说明我查过一些,但没有查到太多有用的信息,连百度#include <TH/TH.h>都百度不 ...

Thu Aug 15 06:04:00 CST 2019 3 475
学习Faster R-CNN代码roi_pooling(二)

roi_pooling理解起来比较简单,所以我就先看了一下这部分的代码。 roi_pooling目录下 -src文件夹下是c和cuda版本的源码。 -functions文件夹下的roi_pool.py是继承了torch.autograd.Function类,实现RoI层的foward ...

Thu Aug 15 05:54:00 CST 2019 0 693
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM