Tensorflow在手机端的部署——官网Android工程源码分析之TensorFlowYoloDetector.java (1)

文章分析下tensorflow提供的官方Android工程的源码分析,后续涉及更改代码,因此有必要对其做深入理解。

首先工程文件路径为:tensorflow-master\tensorflow\examples\android

由于这个android工程中实现了目标检测,风格迁移,语音,图像分类四个功能,其中目标检测中有用到yolo检测,有用到ssd-mobilenet v1检测,还有就是用到multi-box做检测。本文将针对yolo做检测需要用到的TensorFlowYoloDetector.java代码部分进行详细讲解。

 

 

其中yolo v2在20类训练后得到的output上得到的结果按下图次序进行排列:

即【第一个框:x,y,w,h,confidence,class1,……,class20】【第二个框:x,y,w,h,confidence,class1,……,class20】……【第13x13x5个框:x,y,w,h,confidence,class1,……,class20】

Tensorflow在手机端的部署——官网Android工程源码分析之TensorFlowYoloDetector.java (1)_第1张图片

 

TensorFlowYoloDetector.java部分代码如下:

/* Copyright 2017 The TensorFlow Authors. All Rights Reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use 

你可能感兴趣的:(Android,TensorFlow,深度学习/机器学习,移动端(边缘设备)深度学习,Android,TF)