diff --git a/examples/onnx/yolov5/test.py b/examples/onnx/yolov5/test.py index a1c9988..f7ce11e 100644 --- a/examples/onnx/yolov5/test.py +++ b/examples/onnx/yolov5/test.py @@ -240,7 +240,7 @@ if __name__ == '__main__': # pre-process config print('--> Config model') - rknn.config(mean_values=[[0, 0, 0]], std_values=[[255, 255, 255]]) + rknn.config(mean_values=[[0, 0, 0]], std_values=[[255, 255, 255]], target_platform='rk3588') print('done')