MMDETECTION 使用步驟
-
建立voc資料集
參考鏈接：https://blog.csdn.net/weixin_42447868/article/details/105675158 -
下載預訓練權重
建立mmdetection/checkpoints目錄，下載對應的預訓練權重,
權重檔案鏈接
（1）configs對應網路里的README.md 進入后可以通過cfg檔案驗證對應的weight
（2）https://mmdetection.readthedocs.io/zh_CN/v2.19./model_zoo.html#id2 -
修改組態檔
3.1mmdet/core/evalution/class_names.py #voc_classes 修改類別
3.2mmdet/datasets/voc.py #一個類要加 逗號，修改類別
3.3config/ base/models/faster_rcnn.py
1.num_classes
2.checkpoint=checkpoints/…pth #使用預訓練模型，或者3.5.5load_from 中修改
3.4config/ base/datasets/voc0712.py
1.注釋掉line41中，voc2012，
2.修改訓練、驗證、測驗 txt路徑
3.samples_per_gpu #每個gpu檢測幾張圖片，顯卡記憶體不足可以調小
workers_per_gpu #使用幾個執行緒
4.修改影像尺寸 img_scale , 顯卡記憶體不足可以調小
5.pipeline 影像在線增強策略
3.5config/default_runtime.py
1.checkpoint_config #多少epoch 保存一次模型
2.log_config #多少次迭代 輸出一次日志
3.hooks=[ dict(), dict(type=‘TensorboardLoggerHook’)] #使用tensorboard
4.log_level #輸出日志等級
5.load_from # 加載lastest.pth 繼續訓練
6.resume_from #斷點繼續訓練
tips：checkpoints、load_from、resume_from優先級：mmdet默認加載權重優先級別是resume_from(斷點加 載)，load_from，pretrained的順序，所以需要從load_from加載預訓練權重
3.6configs/schedule/schedule_1x.py #修改超引數
3.7configs/fasterrcnn/fasterrcnn_r50_fpn_1x_coco.py
修改路徑 除了base外的注釋掉
4.需要重新編譯 python setup.py install
5.訓練 ：引數檔案，gpu，輸出路徑
python tools/train.py configs/faster_rcnn/....py --gpus 1 --work-dir WORK_DIR checkpoints/...pth
6.測驗：引數檔案，最近的權重檔案，輸出路徑
python tools/test.py configs/faster_rcnn/...py work_dir/lastest.pth --show-dir output --out output/result.pkl
7.工具
7.1繪制結果曲線
python tools/analyze/ana.py plot_curve workdir/~/888.log.json --keys loss lr --out loss.pdf
7.2程序可視化 前提： 3.5.3
tensorboard --logdir work_dirs/88/88/ #events的上一層目錄
7.3測驗結果評價 前提：測驗命令加–out work_dirs/result.pkl
python tools/analysis_tools/eval_metric.py work_dirs/faster_rcnn_r50_fpn_1x_voc0712.py output/result30.pkl --eval mAP
#測驗時保存的pkl，測驗時使用的組態檔,需要分析什么資料
tips:1.學習率策略
lr_config = dict(
policy='step', # 優化策略
warmup='linear', # 初始的學習率增加的策略，linear為線性增加
warmup_iters=500, # 在初始的500次迭代中學習率逐漸增加
warmup_ratio=1.0 / 3, # 起始的學習率
step=[8, 11]) # 在第8和11個epoch時降低學習率
————————————————
著作權宣告：本文為CSDN博主「藏云閣主」的原創文章，遵循CC 4.0 BY-SA著作權協議，轉載請附上原文出處鏈接及本宣告，
原文鏈接：https://blog.csdn.net/hajlyx/article/details/85991400
參考鏈接
https://bbs.huaweicloud.com/blogs/detail/198417
https://blog.csdn.net/H_Ben_Ben/article/details/120336630
https://blog.csdn.net/hajlyx/article/details/85991400
https://blog.csdn.net/weixin_36642958/article/details/96905866
https://blog.csdn.net/syysyf99/article/details/96574325
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/382983.html
標籤：其他