fasterrcnn使用

一训练过程

1 准备数据集

数据集格式：VOC

└── VOCdevkit
└── VOC2012
├── Annotations（最后要的格式）
│   ├── 00a1c979-aed0-4765-9c23-bdf8f3f6eb9c.xml
│   ├── 0a02b6c7-88c8-4c47-8ac9-153840af2746.xml
│   ├── 0a0ad157-a13b-4bd5-97f5-5ee21fa9c4db.xml
│   ├── 0a0e9c6b-79e9-41ab-b333-b1d73914b48b.xml
│   └── 0a1ab3f5-6d77-46bc-936e-a5ec55070c3e.xml
├── ImageSets(代码生成，按比例随机分配)
│   └── Main
│    ├── train.txt
│    └── val.txt
├── JPEGImages（原图）
│   ├── 00a1c979-aed0-4765-9c23-bdf8f3f6eb9c.jpg
│   ├── 0a02b6c7-88c8-4c47-8ac9-153840af2746.jpg
│   ├── 0a0ad157-a13b-4bd5-97f5-5ee21fa9c4db.jpg
│   ├── 0a0e9c6b-79e9-41ab-b333-b1d73914b48b.jpg
│   ├── 0a1ab3f5-6d77-46bc-936e-a5ec55070c3e.jpg
│   └── 0a2ebaae-832e-423c-9798-cbb2fbefd85f.jpg
└── labels（只是我的初始数据的格式）
├── 00a1c979-aed0-4765-9c23-bdf8f3f6eb9c.txt
├── 0a02b6c7-88c8-4c47-8ac9-153840af2746.txt
├── 0a0ad157-a13b-4bd5-97f5-5ee21fa9c4db.txt
├── 0a0e9c6b-79e9-41ab-b333-b1d73914b48b.txt
├── 0a1ab3f5-6d77-46bc-936e-a5ec55070c3e.txt
└── 0a2ebaae-832e-423c-9798-cbb2fbefd85f.txt
2 修改训练参数   train_res50_fpn.py

有两个地方需要下载。url分别是：

https://download.pytorch.org/models/resnet50-0676ba61.pth

https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_coco-258fb6c6.pth

修改参数：

运行即可。

训练过程，训练完一个epoch之后会进行验证（显示"Test:"），有ap和ar值。

3 验证 validation.py

用于调用训练好的模型权重去计算验证集/测试集的COCO指标
以及每个类别的mAP(IoU=0.5)

添加json文件并修改

4 预测 predict.py

绿色的都要改，最后生成json文件，并画出预测图

fasterrcnn使用相关推荐

RCNN系列、Fast-RCNN、Faster-RCNN、R-FCN检测模型对比
RCNN系列.Fast-RCNN.Faster-RCNN.R-FCN检测模型对比一．RCNN 问题一:速度经典的目标检测算法使用滑动窗法依次判断所有可能的区域.本文则预先提取一系列较可能是物体的候 ...
基于CNN目标检测方法（RCNN，Fast-RCNN，Faster-RCNN，Mask-RCNN，YOLO，SSD）行人检测，目标追踪，卷积神经网络
一.研究意义卷积神经网络(CNN)由于其强大的特征提取能力,近年来被广泛用于计算机视觉领域.1998年Yann LeCun等提出的LeNet-5网络结构,该结构使得卷积神经网络可以端到端的训练,并应 ...
python脚本——图片重命名、图片合成视频、faster-rcnn画P-R曲线
调试faster rcnn算法实用的python脚本目录调试faster rcnn算法实用的python脚本一.前言二.常用python脚本三.后记一.前言最近在做关于目标检测算法的研究 ...
目标检测之Faster-RCNN的pytorch代码详解(数据预处理篇)
首先贴上代码原作者的github:https://github.com/chenyuntc/simple-faster-rcnn-pytorch(非代码作者,博文只解释代码) 今天看完了simple- ...
fasterrcnn深度学习口罩检测
fasterrcnn深度学习口罩检测前言 FasterRCNN原理详解训练我们自己的FasterRCNN 使用labelimg制作我们自己的VOC数据集 FasterRCNN训练详解源码地址与小 ...
FasterRCNN代码解读
之前的文章简要介绍了Faster-RCNN等物体检测的算法,本文将从代码角度详细分析介绍Faster-RCNN的实现.本文使用的代码参考了chenyuntc的实现,代码的位置看这里.需要注意的是,本文 ...
windows下faster-rcnn遇到的检测框重复很多问题 nms
最近在做目标检测.于是就采用了RBG大神的faster rcnn进行切入.从RCNN开始,一直到Faster rcnn,先把理论过了一遍.接下来就是实践了,准备跑下代码.faster rcnn看了大体 ...
『计算机视觉』经典RCNN_其一：从RCNN到Faster-RCNN
RCNN介绍目标检测-RCNN系列一.目标检测 1.两个任务目标检测可以拆分成两个任务:识别和定位图像识别(classification) 输入:图片输出:物体的类别评估方法:准确率定位 ...
Faster-RCNN 自己的数据训练
参考网址:https://blog.csdn.net/l297969586/article/category/7178545(一呆飞仙)Faster-RCNN_TF代码解读, 参考网址:https:/ ...
图像理解之物体检测object detection，模型rcnn/fastrcnn/fasterrcnn原理及概念
A,https://www.cnblogs.com/zhengzhe/p/7783270.html RCNN选择性搜索(Selective Search) RCNN选择性搜索(Selective Se ...

fasterrcnn使用

fasterrcnn使用相关推荐

最新文章

热门文章