一、数据集准备,使用label标注好本身的数据集。
https://github.com/tzutalin/labelImg 打开链接直接下载数据标注工具,python
二、具体的大师代码见下连接
https://github.com/YunYang1994/tensorflow-yolov3git
三、个人代码训练步骤,这里我使用大神的浣熊数据集进行测试
个人代码链接:干货-https://github.com/wuzaipei/python_TensorFlow_yolov3-github
四、具体步骤
一、 训练数据:python core/convert_tfrecord.py --dataset_txt .\raccoon_dataset\labels.txt --tfrecord_path_prefix .\raccoon_dataset\raccoon_train\wzp_new_train工具
二、更新目标中心点:python kmeans.py测试
三、训练:python quick_train.pyui
四、转换model:python convert_weight.pyurl
五、预测 : python quick_test.pyspa