
AI
wxshan3
这个作者很懒,什么都没留下…
展开
-
SSD: Single Shot MultiBox Detector 模型fine-tune
ssd模型fine-tune的步骤(使用coco数据集预训练)(1)下载coco训练的模型(2)使用convert-model.py提取感兴趣的类型的参数,然后保存参数文件为新的模型文件(3)根据(2)得到的模型文件作为微调训练的预训练模型,下载07+12+coco(4)用下载的模型的finetune_ssd_*.py继续训练模型...原创 2018-09-19 11:04:27 · 201 阅读 · 0 评论 -
opencv3中的单目标跟踪
opencv的3版本以上支持了深度学习的一些东西,其中单目标跟踪就是其中一个,以下代码就是官网上的测试例子#include <opencv2/opencv.hpp>#include <opencv2/tracking/tracking.hpp>#include <opencv2/core/ocl.hpp>#include <opencv2/t...原创 2019-03-20 18:39:03 · 520 阅读 · 0 评论 -
TensorRT 系列之入门篇-优化特点
一 tensorTR对于kernel launch以及每层tensor data读取的速度不对称的问题的解决:(1)(纵向融合)通过融合相同顺序的操作来减少Kernel launch的消耗以及避免层之间的读写操作.如卷积;bias和Relu层可以融合成一个kernel,这里称之为CBR(2)(横向融合)TensorRT会去挖掘输入数据且filter大小相同但weights不同的层,对于这些层不...翻译 2018-09-20 09:53:19 · 1355 阅读 · 0 评论 -
yolo的参数解读
一、网络结构图:layer filters size input output 0 conv 32 3 x 3 / 1 416 x 416 x 3 -> 416 x 416 x 32 1 max 2 x 2 / 2 416 x 416 x 32 -...原创 2018-09-19 14:29:12 · 1519 阅读 · 0 评论 -
将检测的物体转换成xml文件
# -*- coding: utf-8 -*-import reimport shutilimport stringimport numpy as npfrom skimage import ioheadstr = """\<annotation> <folder>VOC2007</folder> <filename&g...原创 2018-09-19 14:23:34 · 530 阅读 · 0 评论 -
将caffemodel的参数提取出来,转换成json文件
以下python代码可以将caffemodel的参数转换成json文件import osimport sysimport jsonimport argparseimport tempfileimport subprocessfrom google.protobuf.descriptor import FieldDescriptor as FDcaffe_root = '../...翻译 2018-09-19 14:20:54 · 840 阅读 · 1 评论 -
SSD: Single Shot MultiBox Detector in TensorFlow
SSD: Single Shot MultiBox Detector in TensorFlow(1)下载tensorflow的ssd的模型代码(2)将./checkpoint下的模型压缩文件解压出来,命令为: cd ./checkpoint unzip ssd_300_vgg.ckpt.zip(3)制作数据集,创建bash文件,将下面的代码放入,接着运行bash文件 ...翻译 2018-09-19 11:27:36 · 258 阅读 · 0 评论 -
yolo训练自己的模型
一、将标注好的图片生成可识别的格式: 1.创建文件夹,结构按照voc的结构 2.将原始照片放在JPEGImages里面,将xml文件放在Annotations里面。 3.运行create_list.py将图片分成test,val,train三个部分,同时在ImageSets的Main里面生产对应的三个.txt文件。 4.运行my_voc_label.py生产...原创 2018-09-19 11:15:53 · 547 阅读 · 0 评论 -
SSD训练自己的数据
一、训练自己的数据: (1)在/data目录下创建一个自己的文件夹: cd /data mkdir mydataset (2)把/data/VOC0712目录下的create_list.sh 、create_data.sh、labelmap_voc.prototxt 这三个文件拷贝到/mydataset下: cp data/crea...原创 2018-09-19 11:11:56 · 285 阅读 · 0 评论 -
使用队列来缓冲视频流和神经网络
在使用神经网络进行视频的结构化的时候,当视频流的个数足够大的时候,需要使用队列来起到缓冲的作用。下面的代码就是我实现的,在SSD模型检测时,视频流的个数多的时候,视频流产生数据速度远远高于检测图片的速度,所以需要队列#include <caffe/caffe.hpp>#ifdef USE_OPENCV#include <opencv2/core/core.hpp>...原创 2019-03-20 20:12:15 · 1178 阅读 · 3 评论