本文整理汇总了Python中modeling.model_builder.Generalized_RCNN方法的典型用法代码示例。如果您正苦于以下问题：Python model_builder.Generalized_RCNN方法的具体用法？Python model_builder.Generalized_RCNN怎么用？Python model_builder.Generalized_RCNN使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类modeling.model_builder的用法示例。
在下文中一共展示了model_builder.Generalized_RCNN方法的3个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: initialize_model_from_cfg
# 需要导入模块: from modeling import model_builder [as 别名]
# 或者: from modeling.model_builder import Generalized_RCNN [as 别名]
def initialize_model_from_cfg(args, gpu_id=0):
"""Initialize a model from the global cfg. Loads test-time weights and
set to evaluation mode.
"""
model = model_builder.Generalized_RCNN()
model.eval()
if args.cuda:
model.cuda()
if args.load_ckpt:
load_name = args.load_ckpt
logger.info("loading checkpoint %s", load_name)
checkpoint = torch.load(load_name, map_location=lambda storage, loc: storage)
net_utils.load_ckpt(model, checkpoint['model'])
if args.load_detectron:
logger.info("loading detectron weights %s", args.load_detectron)
load_detectron_weight(model, args.load_detectron)
model = mynn.DataParallel(model, cpu_keywords=['im_info', 'roidb'], minibatch=True)
return model 示例2: initialize_model_from_cfg
# 需要导入模块: from modeling import model_builder [as 别名]
# 或者: from modeling.model_builder import Generalized_RCNN [as 别名]
def initialize_model_from_cfg(args, gpu_id=0):
"""Initialize a model from the global cfg. Loads test-time weights and
set to evaluation mode.
"""
model = Generalized_RCNN()
model.eval()
if args.cuda:
model.cuda()
if args.load_ckpt:
load_name = args.load_ckpt
logger.info("loading checkpoint %s", load_name)
checkpoint = torch.load(load_name, map_location=lambda storage, loc: storage)
net_utils.load_ckpt(model, checkpoint['model'])
# model.load_state_dict(checkpoint['model'])
if args.load_detectron:
logger.info("loading detectron weights %s", args.load_detectron)
load_detectron_weight(model, args.load_detectron)
model = mynn.DataParallel(model, cpu_keywords=['im_info', 'roidb'], minibatch=True)
return model 示例3: initialize_model_from_cfg
# 需要导入模块: from modeling import model_builder [as 别名]
# 或者: from modeling.model_builder import Generalized_RCNN [as 别名]
def initialize_model_from_cfg(args, roidb=None, gpu_id=0):
"""Initialize a model from the global cfg. Loads test-time weights and
set to evaluation mode.
"""
model = model_builder.Generalized_RCNN()
model.eval()
cfg.immutable(False)
cfg.TEST.CLASS_SPLIT = {'source': roidb[0]['source'], 'target': roidb[0]['target']}
cfg.immutable(True)
if 'word_embeddings' in roidb[0]:
model.Box_Outs.set_word_embedding(torch.tensor(roidb[0]['word_embeddings']))
if cfg.MODEL.IGNORE_CLASSES:
if cfg.MODEL.IGNORE_CLASSES == 'all':
roidb[0]['all'] = roidb[0]['source'] + roidb[0]['target']
model._ignore_classes = roidb[0][cfg.MODEL.IGNORE_CLASSES]
model.Box_Outs._ignore_classes = roidb[0][cfg.MODEL.IGNORE_CLASSES]
if True:
tmp = {}
for rel in roidb[0]['relationships']:
tmp[(rel['subject_id'], rel['object_id'])] = \
tmp.get((rel['subject_id'], rel['object_id']), []) + [rel['rel_id']]
if cfg.MODEL.RELATION_COOCCUR:
for k in tmp:
tmp[k] = [1]
if cfg.MODEL.NUM_RELATIONS > 0:
model.Rel_Outs.relationship_dict = tmp
if args.cuda:
model.cuda()
if args.load_ckpt:
load_name = args.load_ckpt
logger.info("loading checkpoint %s", load_name)
checkpoint = torch.load(load_name, map_location=lambda storage, loc: storage)
net_utils.load_ckpt(model, checkpoint['model'])
if args.load_detectron:
logger.info("loading detectron weights %s", args.load_detectron)
load_detectron_weight(model, args.load_detectron)
model = mynn.DataParallel(model, cpu_keywords=['im_info', 'roidb'], minibatch=True)
return model