TensorRT-RepVGG:TensorRT实现的“ RepVGG

时间:2021-05-24 13:30:51
【文件属性】:
文件名称:TensorRT-RepVGG:TensorRT实现的“ RepVGG
文件大小:7KB
文件格式:ZIP
更新时间:2021-05-24 13:30:51
C++ RepVGG 来自“ RepVGG:使VGG风格的ConvNets再次伟大”的RepVGG模型 有关Pytorch的实现,可以参考 怎么跑 生成wts文件。 git clone https://github.com/DingXiaoH/RepVGG.git cd RepVGG 您可以使用以下方法将训练后的模型转换为推理时间结构: python convert.py [weights file of the training-time model to load] [path to save] -a [model name] 例如, python convert.py RepVGG-B2-train.pth RepVGG-B2-deploy.pth -a RepVGG-B2 然后将TensorRT-RepVGG/gen_wts.py复制到RepVGG并生成.wts文件,例如
【文件预览】:
TensorRT-RepVGG-master
----gen_wts.py(895B)
----logging.h(1KB)
----CMakeLists.txt(685B)
----LICENSE(1KB)
----README.md(1000B)
----repvgg.cpp(13KB)

网友评论