1、數據集准備,使用label標注好自己的數據集。
https://github.com/tzutalin/labelImg 打開連接直接下載數據標注工具,
2、具體的大師代碼見下鏈接
https://github.com/YunYang1994/tensorflow-yolov3
3、我的代碼訓練步驟,這里我使用大神的浣熊數據集進行測試
我的代碼連接:干貨-https://github.com/wuzaipei/python_TensorFlow_yolov3-
4、具體步驟
1、 訓練數據:python core/convert_tfrecord.py --dataset_txt .\raccoon_dataset\labels.txt --tfrecord_path_prefix .\raccoon_dataset\raccoon_train\wzp_new_train
2、更新目標中心點:python kmeans.py
3、訓練:python quick_train.py
4、轉換model:python convert_weight.py
5、預測 : python quick_test.py