加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
convert_trt.sh 183 Bytes
一键复制 编辑 原始数据 按行查看 历史
xiaowei 提交于 2022-10-05 22:57 . add recognition
#!/bin/bash
if [ ! -f $1 ] ; then
echo "usage: $0 model.onnx"
exit 1
fi
output=$(echo $1 |sed s/onnx$/engine/)
onnx=$1
shift
trtexec --onnx=${onnx} --saveEngine=${output} --fp16 $*
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化