使用Tensorflow进行目标检测

很久以前我训练了一个目标检测AI模型

目录

在一个寒冷的七月冬天……
那是在澳大利亚……
我感到迫切需要训练一个AI模型,用于检测未封口的混凝土钢筋……

所以我从当地的Bunnings Warehouse商店购买了一些新的闪亮钢筋,比如这些:

混凝土钢筋

以及一些钢筋安全帽,比如这些:

混凝土钢筋帽

从帽子开始

我用这些黄色钢筋帽拍了100张不同构图的照片,并用LabelImg进行标注,接着使用了一个SSD模型,然后又用EfficientNet进行了一些在Python中的编码,模型很快就学会了我想要的东西。

钢筋帽 - 左手

钢筋帽 - 右手

起初,模型几乎将所有黄色圆柱形物体都识别为钢筋帽:

咖啡杯帽

而且边界框可以更精确一些……

钢筋随后到来

我从车库里拿了一些其他有用的工具,又拍了100多张类似的照片,模型开始识别得更好了……

车库工具目标检测

是的,确实更好了。

车库工具目标检测

在车库里进行AI目标检测训练

移动应用

最后,我训练了一个小巧的SSD AI模型,可以在手机上运行,几乎没有延迟,创建了一个简单的Android应用,这是它在三星S8上的样子:

手机上的SSD AI

手机上的目标检测

手机上的TensorFlow

这是一次非常成功的实验。

有用的链接