安全帽图像识别python_基于Faster R-CNN的安全帽目标检测

公众号关注 “Python遇见机器学习”

设为“星标”，第一时间知晓最新干货~

来源：机器学习AI算法工程

训练模型前的准备

A.数据准备

数据的标注仍然采用VOC格式的数据标注形式，如果是其他的标注形式比如COCO请自行实现相关代码。将数据最终转化为如下形式：

path_filename, x1, y1, x2, y2, class_name

格式说明

可以运行如下代码实现数据集的准备工作:

python3 ./data/data_pro.py

将在./data文件夹下生成annotation.txt文件，这样训练数据的准备工作即完成。

B.配置文件准备

根据自己的训练集和训练任务修改./keras_frcnn/config.py的配置文件,相关参数的解释和配置如下：

训练模型

预训练模型的下载地址：https://github.com/fchollet/deep-learning-models

Shell下运行：

python3 train_frcnn.py --path="./data/annotation.txt" --network="vgg" --input_weight_path=&#