Retinaface訓練自己的資料集 記錄

2021-10-25 20:25:52 字數 1021 閱讀 5756

retinaface訓練自己的資料集 記錄

win 10

mxnet

python 3.6

ps: 由於新手習慣了win10和tf,沒用過linux以及pytorch,所以在跑通insightface也走了不少彎路。

pip清華源安裝pip install mxnet_cu90mkl -i 我用的gpu,可以設定 os.environ[『cuda_visible_devices』]=『0』,然後這個問題就不報錯了

雖然有gpu但cuda不匹配,在製作好資料集跑時也遇到了很多問題,花了很久才解決

改了train.py 以及 multiboxloss.py中的帶cuda的,替換成cpu

解決assertionerror:torch not compiled with cuda enabled報錯問題

把自己的資料集的原圖放在 images

以及做好的label

ps: 有關label.txt

(因為空格和路徑的問題一直出現 報錯height, width, _ = img.shape attributeerror: 『nonetype』 object has no attribute 『shape』

第二行前四個分別表示臉部的位置資訊

b0=xmin

b1=ymin

b2=xmax-xmin

b4=ymax-ymin

第五個至第十九個數值表示人臉的五個關鍵點的座標,如果不需要,可以全部標-1

最後乙個置信度

(有位大神告訴我說這個是根據現有的人臉檢測演算法測試出來的,也可以根據這個值設定是否將其加入訓練)

在config設定相關引數,改train中的各項路徑就可以開始訓練了

訓練自己的資料

為了完成畢業設計,還需要繼續學習caffe,並能夠成功地訓練自己的實驗資料,這裡整理caffe整個訓練過程。使用caffe訓練卷積神經網路主要包括以下幾個步驟 1 預處理影象,一般是指將影象轉化為具有同樣大小的畫素 2 根據設計好的網路結構配置相應的網路結構檔案和訓練引數檔案 3 訓練模型並測試結果...

refinedet pytorch訓練自己的資料

方便起見便按照voc資料格式修改,並修改 中關於voc資料使用的地方。1 將自己的資料製作成voc格式,可參考 2 data config.py 2.1 修改home 自己的資料集路徑 2.2 修改voc資料訓練 為了方便檢視改動,用紅色字型標註 voc 512 3 data voc0712.py ...

caffe訓練自己的資料

本文介紹如何使用 caffe 對自己的影象資料進行分類。1 資料庫準備 由於資料收集比較費時,為了簡單說明,我用了兩類,dog和 bird 每種約 300張。train200 張,val100張。新建乙個資料夾 mine 放自己的資料,在 mine 資料夾下新建 train 和val 資料夾,tra...