在Tensorflow上使用神经网络(SSD)进行实时钉子检测。
此回购类似于。 我使用MobileNetV1-FPN-SSD训练了指甲检测器,并且我使用的数据集由实验室成员标注。
这是作用中的指甲检测器,并画出了线:
带注释的示例如下:
整个过程分为3个步骤:
直方图均衡
肤色二值化(将RGB通道转换为YCbCr并使用inRange方法进行二值化)
MobileNet-SSD-FPN
MobileNetV1-FPN-SSD体系结构可能如下所示:
我先使用MobileNet-SSD,但钉子太小了,无法检测到。 因此,我们可以看到MobileNet-SSD-FPN比MobileNet-SSD更好。
方法
地图
mAP(大)
mAP(中)
mAP(小)
移动网
45.20
62.58
48.14
13.14
75.79
45.81
MobileNet-SSD
2021-09-08 10:19:04
130.83MB
Python
1