- 博客(5)
- 收藏
- 关注
原创 处理数据的代码
记录一些用来处理数据的代码,便于使用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关注的人