修改原始碼
測試訓練自己的資料集
github: ssd: single shot multibox object detector, in pytorch
git clone
coco
sh data/scripts/coco2014.sh
voc 2007 & 2012
sh data/scripts/voc2007.sh
sh data/scripts/voc2012.sh
執行指令碼後會把資料集存放在~/data中
由於pytorch版本差異，需要對原始碼幾處地方進行修改：
調換第97,98行：
loss_c = loss_c.view(num,-1
) loss_c[pos]=0
# filter out pos boxes for now
n = num_pos.data.
sum(
).double(
) loss_l = loss_l.double(
) loss_c = loss_c.double(
)
# 預訓練權重
mkdir weights
cd weights
wget
# 訓練好的voc權重
wget
Caffe 用自己的資料做訓練
1.1資料集分為train和validate，分別寫到train.txt和val.txt中。1.2將準備好的兩個txt，放到caffe root data myfile caffe root是caffe的根目錄，myfile自己命名 將訓練街和驗證集資料夾放到 img full dir1和 img ...
用mmdetection訓練自己的資料集
接觸目標檢測也有半年時間了，看了一些，但是都是看得馬馬虎虎，這次先用mmdetection跑一下faster rcnn，後面徹底熟悉了，再分析它的執行過程。可能有人說先打牢基礎，看這看那，我認為還是得及時地和實際結合，如果有不對指出，請批評指正。mmdetection是商湯提出來的乙個框架吧，如果只...
用caffe訓練自己的資料集 二
本文主要參考了 五 構建網路 在myfile4下建立乙個.prototxt檔案，來搭建網路。檔名為 myfile4 train test.prototxt 構建的網路內容如下 name myfile4 layer transform param data param layer transform ...