"mmdetection3d/docs/en/advanced_guides/datasets/waymo.md" did not exist on "9c03eaa81d1b4ece4f75bcd13395a00b25c99de0"
panoptic-seg_panoptic-fpn_onnxruntime_dynamic.py 675 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
_base_ = [
    '../_base_/base_panoptic-seg_static.py',
    '../../_base_/backends/onnxruntime.py'
]
onnx_config = dict(
    input_shape=None,
    output_names=['dets', 'labels', 'masks', 'semseg'],
    dynamic_axes={
        'input': {
            0: 'batch',
            2: 'height',
            3: 'width'
        },
        'dets': {
            0: 'batch',
            1: 'num_dets',
        },
        'labels': {
            0: 'batch',
            1: 'num_dets',
        },
        'masks': {
            0: 'batch',
            1: 'num_dets',
        },
        'semseg': {
            0: 'batch',
            2: 'height',
            3: 'width'
        },
    },
)