Skip to content

Instantly share code, notes, and snippets.

@j-adamczyk

j-adamczyk/detectron2_demo_2.py

Last active July 18, 2020 10:14
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save j-adamczyk/b7896ab88582d8ba1e2fd86489de9706 to your computer and use it in GitHub Desktop.
Save j-adamczyk/b7896ab88582d8ba1e2fd86489de9706 to your computer and use it in GitHub Desktop.
Download ZIP
Detectron2 demo article - gist 2
Raw
detectron2_demo_2.py
from detectron2 import model_zoo
from detectron2.config import get_cfg, CfgNode
from detectron2.engine import DefaultPredictor
if __name__ == "__main__":
args: argparse.Namespace = _get_parsed_args()
cfg: CfgNode = get_cfg()
cfg.merge_from_file(model_zoo.get_config_file(args.base_model))
cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url(args.base_model)
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
predictor: DefaultPredictor = DefaultPredictor(cfg)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment