34 lines
1.2 KiB
YAML
34 lines
1.2 KiB
YAML
deploy_cfgs:
|
|
- &ObjectDetection-0
|
|
- configs/mmdet/detection/detection_tensorrt_dynamic-320x320-1344x1344.py
|
|
- configs/mmdet/detection/detection_tensorrt-fp16_dynamic-320x320-1344x1344.py
|
|
- configs/mmdet/detection/detection_tensorrt-int8_dynamic-320x320-1344x1344.py
|
|
- &InstanceSegmentation-0
|
|
- configs/mmdet/instance-seg/instance-seg_tensorrt_dynamic-320x320-1344x1344.py
|
|
- configs/mmdet/instance-seg/instance-seg_tensorrt-fp16_dynamic-320x320-1344x1344.py
|
|
- configs/mmdet/instance-seg/instance-seg_tensorrt-int8_dynamic-320x320-1344x1344.py
|
|
|
|
img_dir: images
|
|
|
|
models:
|
|
-
|
|
name: YOLOV3
|
|
deploy_cfg: *ObjectDetection-0
|
|
model_info:
|
|
-
|
|
model_cfg: ../mmdetection/configs/yolo/yolov3_mobilenetv2_320_300e_coco.py
|
|
checkpoint: checkpoints/yolov3_mobilenetv2_320_300e_coco_20210719_215349-d18dff72.pth
|
|
shape: 320x320
|
|
device: cuda:0
|
|
backend_model: end2end.engine
|
|
-
|
|
name: RetinaNet
|
|
deploy_cfg: *ObjectDetection-0
|
|
model_info:
|
|
-
|
|
model_cfg: ../mmdetection/configs/retinanet/retinanet_r18_fpn_1x_coco.py
|
|
checkpoint: checkpoints/retinanet_r18_fpn_1x_coco_20220407_171055-614fd399.pth
|
|
shape: 800x1344
|
|
device: cuda:0
|
|
backend_model: end2end.engine
|