环境信息:昇腾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算子吗? 为啥会不生效呢?
环境信息:昇腾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算子吗? 为啥会不生效呢?