Views
No views yet
COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml)ResizeShortestEdgeIMS_PER_BATCH=16, BASE_LR=0.001, MAX_ITER=150000, STEPS=(80000, 110000), WARMUP_ITERS=1000val split):1import cv2
2from detectron2.config import get_cfg
3from detectron2.engine import DefaultPredictor
4from detectron2.utils.visualizer import Visualizer
5from detectron2.data import MetadataCatalog
6cfg = get_cfg()
7cfg.merge_from_file("config.yaml")
8cfg.MODEL.WEIGHTS = "model_final.pth"
9cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
10# cfg.MODEL.DEVICE = "cpu" # uncomment for CPU
11predictor = DefaultPredictor(cfg)
12image = cv2.imread("path/to/image.jpg")
13outputs = predictor(image)
14# Optional visualization
15metadata = MetadataCatalog.get("zod_infer")
16metadata.thing_classes = ["Vehicle", "Pedestrian", "VulnerableVehicle"]
17vis = Visualizer(image[:, :, ::-1], metadata=metadata, scale=1.0)
18vis = vis.draw_instance_predictions(outputs["instances"].to("cpu"))
19cv2.imwrite("predictions.jpg", vis.get_image()[:, :, ::-1])