@_LJS detectron中加载预训练模型是根据名字加载各层权重的。这样的话,你要在detectron源码里找到你用的mask-rcnn的全连接层定义的python脚本,比如faster-rcnn-head.py中的2mlp_head函数把原来全连接层的名字(如fc6,fc7)修改了(如fc6-new,fc7-new),再按原来一样加载预训练模型就行了。
detectron框架初探Detectron是Facebook AI研究院(FAIR)于2018年初公开的目标检测平台,包含了大量业内最具代表性的目标检测、图像分割、关键点检测算法。该框架主要基于py...