Skip to content

Latest commit

 

History

History
15 lines (8 loc) · 773 Bytes

README.md

File metadata and controls

15 lines (8 loc) · 773 Bytes

TensorRTx

Fork来源: https://github.com/wang-xinyu/tensorrtx

新增了Nanodet

新增了Yolov5-face

新增了stnet (权重来自paddle)

新增了arcface-r100-glink360, pth权重从官方网盘下载,将gen_wts.py放到insightface/recognition/arcface_torch https://github.com/deepinsight/insightface/tree/master/recognition/arcface_torch 名称是backbone.pth

新增了trt_pose, 官方项目https://github.com/NVIDIA-AI-IOT/trt_pose, 这里使用PaddlePddle的复现权重,项目地址: https://aistudio.baidu.com/aistudio/projectdetail/3516206

新增了yolov5n-v6-prune, 支持yolov5n-v6模型,以及通道数量自适应的裁剪模型, pytorch和paddle都可以转换权重,Paddle项目地址:https://aistudio.baidu.com/aistudio/projectdetail/3452279