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