classification_rknn-fp16_static-224x224.py 272 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
_base_ = ['./classification_static.py', '../_base_/backends/rknn.py']

onnx_config = dict(input_shape=[224, 224])
codebase_config = dict(model_type='end2end')
backend_config = dict(
    input_size_list=[[3, 224, 224]],
    quantization_config=dict(do_quantization=False))