一、问题
女朋友图像处理大作业是要求做安全帽识别、口罩识别、姿态识别、自然场景文字识别等,选口罩识别的人多,就选择安全帽识别吧。
二、分析
利用ssd-mobilenet来做安全帽检测,正好数据集也是voc格式的,只需要配置中更改分类的标签就好。
三、代码
1.数据集处理
删除PartA的图片,对xml格式内容修改
2.分类更改
NUM_CLASSES改成3
class_name更改
3.具体代码
(待补充)GitHub
4.测试代码
结果样例
女朋友图像处理大作业是要求做安全帽识别、口罩识别、姿态识别、自然场景文字识别等,选口罩识别的人多,就选择安全帽识别吧。
利用ssd-mobilenet来做安全帽检测,正好数据集也是voc格式的,只需要配置中更改分类的标签就好。
删除PartA的图片,对xml格式内容修改
NUM_CLASSES改成3
class_name更改
(待补充)GitHub
结果样例