1、拍摄视频文件或者照片,视频文件可以采用代码抽帧或者视频播放器抽帧的方法获取图片。本例采用完美解码的potplay,定时4s,抽取一帧。软件可在官网下载。
2、标注采用精灵标注助手,可在官网上下载。软件选择位置标注,按提示创建要标注的项目值,开始标注,标注完成后,点击下方蓝色对勾,保存标注值,全部标记完成后,导出选择xml格式。
3、拷贝图片和导出的label到想使用的模型下边。
4、修改对应配置文件:
生成csv:修改目录名称为labels,屏蔽不使用的文件夹
添加item:在train_labels_map.pbtxt中添加新增标的物。
5、运行生成csv脚本:BatchXmlToCsv.sh
如果提示ModuleNotFountdError:No module named 'pandas'
使用命令安装:pip install pandas_datareader 相应模块即可。
提示:No module named 'object_detection'
找到3rd_part目录,进入目录,运行sh run.sh,安装第三方库。
最终在data目录下生成.csv 和.record文件
6、修改pipeline.config文件
修改训练参数值。
7、运行Training脚本
提示No module named matplotlib
使用pip3 install matplotlib
提示没有找到tensorflow 模块
拷贝欲使用的模型 model.ckpt相关模块到trainning文件夹