- 博客(5)
- 收藏
- 关注
RSS订阅原创 处理数据的代码
记录一些用来处理数据的代码，便于使用voc数据集import osimport pprint #此方法用来统计xml中不同类别的个数def tongji_xml(Anno_dir): counts = {} for filename in os.listdir(Anno_dir): tree = read_xml(os.path.join(Anno_dir,filename)) text_nodesCate = [] for child in tree.
2020-12-08 14:11:45
313
原创 使用CenterNet训练自己的数据集
最近在做目标检测的项目，使用YOLOv3，Mask R-CNN，Faster R-CNN训练了自己的数据，效果还可以，在保证精度的情况下，速度有点慢。于是就开始找别的方法，发现了2019年CVPR的文章CenterNet :Objects as Points，想尝试看看效果。于是就开始了艰难的配置，在这里记录一下。我是在服务器上训练的，环境是cuda10 ,使用的代码是https://githu...
2019-08-06 11:11:45
7168
22
原创 根据图像制作目标检测的视频
这两天在做目标检测项目，为了帮助理解，需要制作检测视频。在这里记录一下：1.把一系列图片放入文件夹下，把文件夹下所有图片等比例resieze到相同尺寸，比如1300*1300，不足的地方用白色替代。#coding:utf-8import osimport cv2import globimport numpy as npfrom PIL import Image#把图片resize到...
2019-08-06 09:51:18
307
原创 VOC数据集转COCO数据集
这几天为了在maskrcnn_benchmark中训练模型，需要把自己的数据集（VOC）转为COCO数据集的格式，过程中遇到了很多问题。在此记录一下整个过程。1.下载COCOAPI，Matlab安装gcc，因为我是在windows下转换的，所以需要自己安装gcc，听说好像Linux下转换不需要安装；2.matlab中设置工作路径：3.进入Visual Object Classes Cha...
2019-07-18 10:23:05
1268
原创 记录一下这几天训练Faster R-CNN，Mask R-CNN，YOLOv3踩的坑
这几天在训练模型的时候遇到了很多问题，通过上网查资料等方式解决了，在此记录一下：1、Faster R-CNN:使用代码 https://github.com/smallcorgi/Faster-RCNN_TF出现问题：TypeError: argument of type ‘NoneType’ is not iterable.解决办法：把config中的路径改为自己的vgg_16.c...
2019-07-18 10:06:55
1210
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人