AIY视觉套件上的物体检测

时间:2018-08-14 14:19:40

标签: tensorflow tensorflow-lite

我正在研究AIY视觉套件,并尝试在视频流中实现对象检测。 我尝试遵循AIY网站(https://aiyprojects.withgoogle.com/)上发布的方法,但是当我尝试使用引擎盖模型编译器转换模型时似乎遇到了问题。 这是我遵循的步骤的简要摘要:

  1. 为训练集中的每个图像创建xml格式(256x256)的注释。
  2. 将所有图像转换为256x256。
  3. 创建tf记录。
  4. 下载了ssd_mobilnet_v1_coco.config的基本模型并进行了必要的更改(设置num classes = 2,以及训练和val tf记录的路径)。
  5. 训练模型
  6. 将冻结的图导出为binaryproto格式。 通过以下链接可以看到这5个步骤,使用了相同的图像,但调整了大小。(https://medium.freecodecamp.org/trackin ... c86419225e) 所有功能(如创建tf记录和其他相关功能)都可以在此git repo https://github.com/bourdakos1/Custom-Object-Detection
  7. 中找到

导出模型后,使用引擎盖模型编译器时出现以下错误

  

2018-08-14 10:39:19.285588:我   外部/org_tensorflow/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39]   删除未使用的操作之前:556个运算符,893个数组(0个已量化)

     

2018-08-14 10:39:19.302398:我   外部/org_tensorflow/tensorflow/contrib/lite/toco/graph_transformations/graph_transformations.cc:39]   在进行一般图形转换之前:556个运算符,893个数组(0   量化)

     

2018-08-14 10:39:19.324886:F   外部/org_tensorflow/tensorflow/contrib/lite/toco/tooling_util.cc:628]   检查失败:暗淡> = 1(0比1)

0 个答案:

没有答案