def setup():
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
cfg.MODEL.DEVICE = 'cpu'
add_vovnet_config(cfg)
cfg.merge_from_file('projects/sku110/configs/faster_rcnn_ShuffleNetv2_05_FPNLite_1x.yaml')
cfg.SOLVER.IMS_PER_BATCH = 1
cfg.freeze()
default_setup(cfg, {})
return cfg
def setup():
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
cfg.MODEL.DEVICE = 'cpu'
add_vovnet_config(cfg)
cfg.merge_from_file(
'projects/vovnet_sku110/configs/faster_rcnn_V_19_FPN_1x.yaml')
cfg.freeze()
default_setup(cfg, {})
return cfg
def setup(args):
"""
Create configs and perform basic setups.
"""
cfg = get_cfg()
add_vovnet_config(cfg)
cfg.merge_from_file("../config/faster_rcnn_V_99_FPN_3x.yaml")
cfg.merge_from_file(args.config_file)
cfg.merge_from_list(args.opts)
cfg.freeze()
default_setup(
cfg, args
) # if you don't like any of the default setup, write your own setup code
return cfg
def get_pred_img(dir):
dataset = get_img_dicts(dir)
if DEBUG:
dataset = dataset[:500]
cfg = get_cfg()
add_vovnet_config(cfg)
cfg.merge_from_file('/myspace/one_cls_faster_rcnn_V_99_FPN/config.yaml')
cfg.MODEL.WEIGHTS = '/myspace/one_cls_faster_rcnn_V_99_FPN/model_0099999.pth'
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.9
cfg.TEST.AUG.ENABLED = True
cfg.TEST.AUG.MIN_SIZES = (800, )
cfg.TEST.AUG.MAX_SIZE = 4000
cfg.TEST.AUG.FLIP = False
bbox_net = build_model(cfg)
DetectionCheckpointer(bbox_net).load(cfg.MODEL.WEIGHTS)
bbox_net = GeneralizedRCNNWithTTA(cfg, bbox_net)
inst_ds = infer_img(bbox_net, dataset)
return inst_ds