Commit aab76a49 authored by Ilija Radosavovic's avatar Ilija Radosavovic Committed by Facebook Github Bot

Document model types not supported by infer_simple

Reviewed By: rbgirshick

Differential Revision: D8332717

fbshipit-source-id: a3070cd83855b5c91a3395d8bbe6c4a70774a1b9
parent a04ed18a
...@@ -94,10 +94,17 @@ def parse_args(): ...@@ -94,10 +94,17 @@ def parse_args():
def main(args): def main(args):
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
merge_cfg_from_file(args.cfg) merge_cfg_from_file(args.cfg)
cfg.NUM_GPUS = 1 cfg.NUM_GPUS = 1
args.weights = cache_url(args.weights, cfg.DOWNLOAD_CACHE) args.weights = cache_url(args.weights, cfg.DOWNLOAD_CACHE)
assert_and_infer_cfg(cache_urls=False) assert_and_infer_cfg(cache_urls=False)
assert not cfg.MODEL.RPN_ONLY, \
'RPN models are not supported'
assert not cfg.TEST.PRECOMPUTED_PROPOSALS, \
'Models that require precomputed proposals are not supported'
model = infer_engine.initialize_model_from_cfg(args.weights) model = infer_engine.initialize_model_from_cfg(args.weights)
dummy_coco_dataset = dummy_datasets.get_coco_dataset() dummy_coco_dataset = dummy_datasets.get_coco_dataset()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment