segmentation_rknn-int8_static-320x320.py 211 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
_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]])