BiSeNet
BiSeNet copied to clipboard
onnx到处参数应该是pred,否则trt脚本推理会报错
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插件