Faster RCNN训练FLIR红外线数据集
1. Fater RCNN检测网络下载
网络学习视频
[源码地址]https://github.com/bubbliiiing/faster-rcnn-pytorch
2. FLIR 数据集准备
数据集的具体格式和内容请看
FLIR数据集介绍
在该数据集中提供的annotations
文件为json,需要将其转换为xml,由于我之前使用yolov5
网络训练,所以目前我使用的转换方法是从json转换到txt,目前我从txt进一步转换为xml
json转换为txt
可能有些朋友在转.txt文件中存在很多问题,我直接给大家将我转的传到了网盘,大家可以直接下载
链接:链接:https://pan.baidu.com/s/1H1YNqXWF_Ee632a8E3Tngg
提取码:1c4n
txt转换xml
链接:https://pan.baidu.com/s/1QxbdV0zbzNcQFoo4oQ1fgg
提取码:smua
3. 网络调整
1.准备数据集
我们需要将生成的image图像
和xml标签文件
放置到VOCdevkit->VOC2007->Annotations和JPEGImages中。
2.制作类文件
如图,在model_data中创建flir.txt文件,内容为People, Bicycles, Cars
3. 生成对应训练文件
检查是否有txt文件,如果有删除,需要生成对应FLIR数据集的txt文件
点击voc_annotation.py
,修改里面classes_path = model_data/flir.txt
,查找jpg
修改为jpeg
即可。
点击运行,会生成刚才删除的txt文件,可以打开看看,2007_train.txt中是训练图像的存储地址和标签。
4. 开始训练
点击train.py
文件,修改classes_path = 'model_data/flir.txt'
。点击运行按钮开始训练。
5. 预测
将val验证集的图像放入
修改预测的权重文件(选择loss和val-loss都小的权重值即可),修改类文件位置。
更改输入文件名称,复制改成导入的文件名称。
运行predict.py
程序,生成img_out图像。
检测结果如下图所示:
总结
对比我之前训练的yolov5来说,这个检测效果感觉没那么好,但是相对于纯视觉来说已经好多了,削弱了光线的影响。
最后,我发现我训练的结果中自行车Bicycles
的检测有点问题,如果有博友按照这个方法训练完成了,可以看看Bicycles
类能不能正常检测。
Faster RCNN训练FLIR红外线数据集相关推荐
- 目标检测simple Faster R-CNN训练自己的数据集
一.复现 刚开始接触目标检测,自己动手复现的第一个开源项目是github上chenyuntc的simple faster rcnn.历经千辛万苦最后貌似因为服务器显卡内存不够,在训练时一直出现一个错误 ...
- faster rcnn 训练自己的数据集---踩坑记录!!!
1.下载代码: git clone https://github.com/jwyang/faster-rcnn.pytorch.git 也可以暴力下载<--直接download压缩包--> ...
- mask rcnn训练自己的数据集
原文首发于微信公众号「3D视觉工坊」--mask rcnn训练自己的数据集 前言 最近迷上了mask rcnn,也是由于自己工作需要吧,特意研究了其源代码,并基于自己的数据进行训练~ 本博客参考:ht ...
- Faster RCNN 训练自己的检测模型
Faster RCNN 训练自己的检测模型 一.准备自己的训练数据 根据pascal VOC 2007的训练数据集基本架构,第一步,当然是要准备自己的训练图片集,本文直接将自己的准备的图片集(.jpg ...
- Fast RCNN 训练自己的数据集(3训练和检测)
Fast RCNN 训练自己的数据集(3训练和检测) 转载请注明出处,楼燚(yì)航的blog,http://www.cnblogs.com/louyihang-loves-baiyan/ https ...
- Faster RCNN 训练中的一些问题及解决办法
今天使用Faster RCNN训练自己的数据的时候,出现了一些因为boost或者是numpy版本不兼容导致的问题,经过各种查资料和求助大神,总算是顺利把网络跑起来了.下面内容都是今天亲测出现的问题并与 ...
- Faster rcnn 训练coco2017数据报错 RuntimeError: CUDA error: device-side assert triggered
Faster rcnn 训练coco2017数据报错 RuntimeError: CUDA error: device-side assert triggered 使用faster rcnn训练自己的 ...
- faster rcnn在自己的数据集上训练
本文是一个总结,参考了网上的众多资料,汇集而成,以供自己后续参考. 一般说来,训练自己的数据,有两种方法:第一种就是将自己的数据集完全改造成VOC2007的形式,然后放到py-faster-rcnn/ ...
- Faster RCNN训练自己数据集报错记录及解决
Faster R-CNN源码网址:https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5点击打开链接 环境:python3.6 tenso ...
最新文章
- 吴恩达老师深度学习视频课笔记:构建机器学习项目(机器学习策略)(1)
- win10前面板耳机没声音
- 【Java核心面试宝典(2),记一次美团Java研发岗的面试经历
- BPI:bit for Webduino WEB:Bit 教育版平台正式发布,支持离线安装使用
- python爬取抖音用户数据_python批量爬取下载抖音视频
- idea去除重复代码校验
- 常用的函数式接口_Function接口练习_自定义函数模型拼接
- mysql 4.0手工注入_手工注入——MySQL手工注入实战和分析
- listary什么意思_listary pro有什么功能
- 《Redis开发与运维》笔记-复制
- 什么是IS-IS中间系统到中间系统?网工、运维必看
- SQL server 2005下载地址
- c语言中文网GUI,golang gui介绍
- 小米路由做文件共享服务器,群晖NAS+小米路由器文件直接传输
- 例3.1 有人用温度计测量出用华氏法表示的温度,今要求把它转换为以摄氏法表示的温度。
- MAC地址IP地址 端口
- Android2.1源码目录结构
- 无法找到指定DLL库文件“v8.dIl“中的输出命令“Initializev8”
- Error during artifact deployment. See server log for details.解决方法
- mc服务器怎么修改祭坛的概率,邪术祭坛 (Eldritch Altar)
热门文章
- 【心理】执行功能障碍
- 百度地图-初步的地图创建和标识
- 熊猫烧香.威金.落雪.SXS.ARP.网络执法管.AUTORUN.INF等高危病毒清除
- 幅相曲线渐近线_幅相曲线.ppt
- 加州欧文计算机工程专业,加州大学欧文分校计算机工程排名第29(2018年TFE美国排名)...
- 世界杯“诈骗杯”?小心这些就对了
- vue图片加载完成前增加loading效果
- 系统管理员设置了系统策略,禁止进行此安装。解决方案
- 统计学 假设检验 P值
- 计算机专业大专考试题,计算机大专考试试题1.doc