文章分析下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】
TensorFlowYoloDetector.java部分代码如下:
/* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use