針對有一點mmdetction基礎的,然后想根據自己的數據集,熟練訓練自己的模型。需要改成自己配置的地方,我會在代碼中做好標記,方便修改。 我們先了解一下mmdetection的基本流程,你想訓練一個模型,你只需要准備的是:數據集,mmdetection的配置文件。 下面我分為兩部分,分別 ...
. 首先確保不加apex前模型能訓練起來,精度正常 . 在環境上安裝好apex: . 修改模型初始化過程,加上apex初始化 修改文件:mmdet tools train.py 導入apex包 修改amp初始化: 文件修改diff如下: . 修改梯度反向寫法 首先弄清楚mmdetection反向的過程,這部分是調用的mmcv實現 runner中實現了訓練的過程,以最基本的runner為例,文件: ...
2021-05-16 21:39 4 761 推薦指數:
針對有一點mmdetction基礎的,然后想根據自己的數據集,熟練訓練自己的模型。需要改成自己配置的地方,我會在代碼中做好標記,方便修改。 我們先了解一下mmdetection的基本流程,你想訓練一個模型,你只需要准備的是:數據集,mmdetection的配置文件。 下面我分為兩部分,分別 ...
v100)。於是開始懷念我熟悉的pytorch+apex,又斷斷續續花了兩天多的時間把整個訓練框架用 ...
關鍵機制:Register注冊,HOOK機制,Runner機制 構建Hook這個類,里面有很多函數。創建Hook這個類的一個對象(包含所有函數)。構建Runner這個類和它的一個對象。向Runner這個對象的self的_hook添加對象。利用函數 ...
由於網絡優化器使用的是目前各個激光網絡開源庫Second、Openpcdet等使用的adam_onecycle 無法使用apex.amp進行初始化,應該是無法識別優化器的問題 怎么都無法解決這個問題,最終決定放棄 后面會嘗試將torch代碼轉成pytorch-lightning試試 ...
mmdetection中的模型checkpoints是需要自己手動下載的,下載步驟如下; 打開mmdetection, 進入configs目錄,可以看到這里面有很多以目標檢測模型命名的文件夾,選擇你想使用的模型,打開目錄,下面以faster_rcnn為例 進入faster_rcnn目錄 ...
一、安裝 https://github.com/open-mmlab/mmdetection/blob/master/docs/INSTALL.md 二、訓練自己的數據 1、數據 mmdet的默認格式是coco的,這里就以voc格式為例,data下文 ...
📖 首先需要准備好數據集,這里有xml標簽數據轉voc數據集格式的說明以及免費分享的數據集:xml轉voc數據集 - 一屆書生 - 博客園 (cnblogs.com) 1. 准備工作目錄 我們的工作目錄,也就是mmdetection目錄,如下所示: configs ...
使用mmdetection訓練自己的VOC數據集 首先,是按照官方文檔組織VOC的數據集結構; 1.目錄結構 mmdetection ├── mmdet ├── tools ├── configs ├── data #手動創建data、VOCdevkit、VOC2007 ...