Skip to content

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

@zgf-suda

Description

@zgf-suda

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插件

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions