|
yolov5使用的训练代码取自ultralytics/yolov5 master分支。
环境:PC ubuntu18.04 x86_64
rknn_tool v1.7.0
python3.6.9环境
tensorflow-gpu==1.14.0
torch==1.2.0
onnx==1.6
torchvision == 0.4.0
插入TB-RK1808S0
用yolov5 onnx轉檔rknn時, 設定如下
rknn.config(reorder_channel='0 1 2',
mean_values=[[0, 0, 0]],
std_values=[[255, 255, 255]],
optimization_level=3,
target_platform = 'rk1808',
output_optimize=1)
ret = rknn.build(do_quantization=QUANTIZE_ON, dataset=DATASET)
轉檔都沒有問題,但是使用一張題圖片做quantization 跟使用2000張圖片做quantization
轉檔出來的model ,同樣測試相同的2000張jpg
測試的結果準確度卻差異不大
請問是有甚麼需要注意的問題? 或是官方有甚麼說明嗎?
謝謝
|
|