realzheng / mini_SSD

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

基于SSD的嵌入式目标检测

1、该模型针对嵌入式设计在小物体检测上的性能会比较好,比如西北工业大学的飞机数据集上(稀疏小目标)。

2、在检测新长宽比先对粗糙大分辨率使用通道添加注意力下采样层目标上性能提升高(多尺度:操场,飞机,过道)。

3、我们模型先对多源粗糙大分辨率使用分离卷积结构下采样层更加速度快。

4、上采样采取标签平滑和训练初始化权重导致损失变换修改让框画的更加准确

VOC Dataset

VOC2007

Training

在模型根目录底下有个setup.py
终端执行   python setup.py  build
再执行   python  setup .py install

在slim文件下同样有setup.py

终端执行   python setup.py  build
再执行   python  setup .py install

Evaluation

Installation

python export_inference_graph.py --input_type image_tensor --pipeline_config_path ./rfcn_resnet101_coco.config 
--trained_checkpoint_prefix ./models/train/model.ckpt-5000 --output_directory ./fine_tuned_model

Citing mini_SSD

 这个项目从工程硬件综合优化的方法来实现移动端高速检测,声明一下这个模型是我参考谷歌的objectdetectionAPI模型的一个子模型和slim方法改过来测试的,主要是为了一个工程产品我们实现嵌入式的使用,考虑到芯片模组之间的计算均衡负担和线程同步问题我们做了相应的设计。

 目前速度和准确率还存在小问题。希望能与大家交流解决这个问题实现95FPS目标,模型最大的优点在速度相对优势的情况下权重体积是普通模型的100/1分之一左右,对普通问题我们采用该模型一般可以控制在5M内实现模型的部署,权重问题一般在2M以内。

 i7CPU的单机PC上测300ms,1200*960尺寸图片。其他相关参数我会补全。最近比较忙大家先自己测测,这个目标的准确率和速度相对可以进一步优化。我会在第二版为大家讲细节,先开源出来解决大家面临模型很大在ARMLinux上的部署问题和产品化自己的idea。模型在使用过程中需要对TensorFlow的源码进行编译和相关ARM上的TF部署。
 
如果你对谷歌的目标检测API使用不熟悉请严格按照我的教程,否则问题很多会导致各种组件之间的编译问题。刚需条件:tensorflow=1.10.0  cuda =8/9训练都可以。

识别结果 识别结果

About


Languages

Language:Python 94.7%Language:Jupyter Notebook 4.2%Language:Shell 0.9%Language:Dockerfile 0.1%