1. 程式人生 > >R-FCN修改訓練自己的資料

R-FCN修改訓練自己的資料

一 train_agnostic_ohem.prototxt

修改為對應的類別:
1
根據設定的anchor個數修改
2
根據設定的anchor個數修改
3
根據設定的anchor個數修改
4
這裡只有兩類,表示前景和背景:
5
根據設定的類別個數修改
6
不用改,只用與區分前景和背景
7
根據設定的類別個數修改
8
不用改,2X4,2是前景和背景,4是迴歸值。
9

二 test_agnostic.prototxt

根據設定的anchor個數修改
11
根據設定的anchor個數修改
12
根據設定的anchor個數修改
13
根據設定的類別個數修改
14
根據設定的類別個數修改
15


不用改,這裡迴歸時並不會區分類別
16
根據設定的類別個數修改,最後輸出的是nx1
17
不用改,2X4,2是前景和背景,4是迴歸值。最後輸出的是 nx4,和faster不一樣,faster會根據類別的個數,每個卡生成四列。
18

三 demo_rfcn.py

從demo就可以看出,作者直接寫的是4:8,和cls_ind並沒有關係。框框都是一樣的,不過是針對每一類的得分會不一樣。
這裡寫圖片描述