BiSeNet

BiSeNet copied to clipboard

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，源码中这样才是会启动自定义的ａｒｇｍａｘ插件

zgf-suda