Views
No views yet
| Metric | Score |
|---|---|
| mAP (0.5:0.95) | 70.0 |
| AP50 | 97.8 |
| AP75 | 84.3 |
| AP (small) | 10.2 |
| AP (medium) | 54.0 |
| AP (large) | 70.8 |
| Class | AP (%) |
|---|---|
| Antelope | 67.4 |
| Lion | 68.8 |
| Elephant | 78.3 |
| Zebra | 66.7 |
| Gorilla | 70.8 |
| Wolf | 72.1 |
| Leopard | 72.1 |
| Giraffe | 64.0 |
1from detectron2.engine import DefaultPredictor
2from detectron2.config import get_cfg
3from detectron2 import model_zoo
4
5cfg = get_cfg()
6cfg.merge_from_file(model_zoo.get_config_file("COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml"))
7cfg.MODEL.WEIGHTS = "path/to/model_final.pth"
8cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
9cfg.MODEL.ROI_HEADS.NUM_CLASSES = 8
10
11predictor = DefaultPredictor(cfg)
12outputs = predictor(your_image)