BiSeNet icon indicating copy to clipboard operation
BiSeNet copied to clipboard

onnx到处参数应该是pred,否则trt脚本推理会报错

Open zgf-suda opened this issue 4 months ago • 0 comments

python tools/export_onnx.py --config configs/bisenetv2_city.py --weight-path /path/to/your/model.pth --outpath ./model.onnx --aux-mode eval,最后的aux-mode应该是pred,源码中这样才是会启动自定义的argmax插件

zgf-suda avatar Aug 28 '25 09:08 zgf-suda