@_LJS detectron中加載預(yù)訓(xùn)練模型是根據(jù)名字加載各層權(quán)重的。這樣的話(huà),你要在detectron源碼里找到你用的mask-rcnn的全連接層定義的python腳本,比如faster-rcnn-head.py中的2mlp_head函數(shù)把原來(lái)全連接層的名字(如fc6,fc7)修改了(如fc6-new,fc7-new),再按原來(lái)一樣加載預(yù)訓(xùn)練模型就行了。
detectron框架初探Detectron是Facebook AI研究院(FAIR)于2018年初公開(kāi)的目標(biāo)檢測(cè)平臺(tái),包含了大量業(yè)內(nèi)最具代表性的目標(biāo)檢測(cè)、圖像分割、關(guān)鍵點(diǎn)檢測(cè)算法。該框架主要基于py...