-
Notifications
You must be signed in to change notification settings - Fork 7
/
detectron2_model_list.py
22 lines (21 loc) 路 1.73 KB
/
detectron2_model_list.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
detectron2_model_list = {
"PascalVOC-Detection/faster_rcnn_R_50_FPN", "PascalVOC-Detection/faster_rcnn_R_50_C4",
"LVISv0.5-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_1x",
"LVISv0.5-InstanceSegmentation/mask_rcnn_R_101_FPN_1x",
"LVISv0.5-InstanceSegmentation/mask_rcnn_R_50_FPN_1x", "COCO-PanopticSegmentation/panoptic_fpn_R_101_3x",
"COCO-PanopticSegmentation/panoptic_fpn_R_50_3x", "COCO-PanopticSegmentation/panoptic_fpn_R_50_1x",
"COCO-Keypoints/keypoint_rcnn_X_101_32x8d_FPN_3x", "COCO-Keypoints/keypoint_rcnn_R_101_FPN_3x",
"COCO-Keypoints/keypoint_rcnn_R_50_FPN_3x", "COCO-Detection/rpn_R_50_FPN_1x",
"COCO-Detection/rpn_R_50_C4_1x", "COCO-Detection/retinanet_R_101_FPN_3x",
"COCO-Detection/retinanet_R_50_FPN_3x", "COCO-Detection/retinanet_R_50_FPN_1x",
"COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x", "COCO-Detection/faster_rcnn_R_101_FPN_3x",
"COCO-Detection/faster_rcnn_R_101_DC5_3x", "COCO-Detection/faster_rcnn_R_101_C4_3x",
"COCO-Detection/faster_rcnn_R_50_FPN_1x", "COCO-Detection/faster_rcnn_R_50_DC5_3x",
"COCO-Detection/faster_rcnn_R_50_DC5_1x", "COCO-Detection/faster_rcnn_R_50_C4_1x",
"COCO-Detection/fast_rcnn_R_50_FPN_1x", "COCO-InstanceSegmentation/mask_rcnn_X_101_32x8d_FPN_3x",
"COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x", "COCO-InstanceSegmentation/mask_rcnn_R_101_DC5_3x",
"COCO-InstanceSegmentation/mask_rcnn_R_101_C4_3x", "COCO-InstanceSegmentation/mask_rcnn_R_50_C4_3x",
"COCO-InstanceSegmentation/mask_rcnn_R_50_C4_1x", "COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_3x",
"COCO-InstanceSegmentation/mask_rcnn_R_50_DC5_1x", "COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_1x",
"COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x", "Cityscapes/mask_rcnn_R_50_FPN"
}