yoloV8推理-置信度过大,求助
收藏回复举报
yoloV8推理-置信度过大,求助
t('forum.solved') 已解决
新人帖
发表于2025-12-17 11:06:04
0 查看

环境信息:昇腾310P3+驱动版本(25.2.0)+CANN版本(8.2.RC1)

1,模型转换 pt------>onnx,代码

onnx_model = model.export(format="onnx", dynamic=False, simplify=True, opset=11)

2, onnx ----> om, 命令:

atc --model=/ultralytics/zwch/electricity/yd_models/yantian/best.onnx --framework=5 --output=yolov1n_atc_zwch --input_shape="images:1,3,640,640"  --soc_version=Ascend310P3 --insert_op_conf=aipp.cfg

aipp.cfg内容:

   matrix_r0c1 : 0

    matrix_r0c2 : 359

    matrix_r1c0 : 256

    matrix_r1c1 : -88

    matrix_r1c2 : -183

    matrix_r2c0 : 256

    matrix_r2c1 : 454

    matrix_r2c2 : 0

    input_bias_0 : 0

    input_bias_1 : 128

    input_bias_2 : 128

    crop: false

    load_start_pos_h : 0

    load_start_pos_w : 0

    crop_size_w : 640

    crop_size_h : 640

    min_chn_0 : 0

    min_chn_1 : 0

    min_chn_2 : 0

    var_reci_chn_0: 0.0039215686274509803921568627451

    var_reci_chn_1: 0.0039215686274509803921568627451

    var_reci_chn_2: 0.0039215686274509803921568627451

}

推理结果(C++)

[INFO]  object detect [72250771588494779397403836416.000000:boat] success

[INFO]  object detect [72250771588494779397403836416.000000:bench] success

[INFO]  object detect [72250771588494779397403836416.000000:microwave] success

[INFO]  object detect [72250771588494779397403836416.000000:bench] success

[INFO]  object detect [72250771588494779397403836416.000000:bench] success

[INFO]  object detect [72250771588494779397403836416.000000:TV monitor] success

[INFO]  object detect [72250700752997536352725630976.000000:sandwich] success

[INFO]  object detect [72249572107408130507519557632.000000:toaster] success

[INFO]  object detect [71939699863535190127887253504.000000:sandwich] success

[INFO]  object detect [71321098188478163822764359680.000000:toothbrush] success

.

.

.

.

.

[INFO]  object detect [5.001038:clock] success

[INFO]  Inference elapsed time : 1.779346 s , fps is 0.562004

[INFO]  Unload model ../model/yolov1n_atc_zwch.om success

[INFO]  destroy context ok

[INFO]  Reset device 0 ok

[INFO]  Finalize acl ok

模型中不是应该有sigmoid算子吗? 为啥会不生效呢?

本帖最后由 匿名用户2025/12/17 14:08:51 编辑

我要发帖子