segmentation_rknn-fp16_static-320x320.py 269 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
9
_base_ = ['./segmentation_static.py', '../_base_/backends/rknn.py']

onnx_config = dict(input_shape=[320, 320])

codebase_config = dict(with_argmax=False)

backend_config = dict(
    input_size_list=[[3, 320, 320]],
    quantization_config=dict(do_quantization=False))