工程化
文章平均质量分 57
今天不标数据的小wu
这个作者很懒,什么都没留下…
展开
-
RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the
如果在GPU上训练的模型,在CPU上推理测试的话,需要在加载模型时使用map_location。如model = torch.load(“model.pth”, map_location=torch.device(“cpu”))推理过程遇到的错误。原创 2023-04-04 12:18:21 · 225 阅读 · 1 评论 -
yolov5m.pt triton部署
1.创建triton docker镜像docker pull nvcr.io/nvidia/tritonserver:21.10-py32.将yolov5m转为onnxA.下载yolov5官方代码B.在运行export.py文件之前,因为onnx必须转成动态的,需要修改dynamic参数,并且根据自己需求修改精度(一般为FP16)。python path/to/export.py --weights yolov5m.pt --img 640 --include onnx --dynamic --原创 2022-01-10 15:30:53 · 2040 阅读 · 10 评论 -
Yolov5m.pt转tensorRT
yolov5m.pt 转TensorRT原创 2021-12-22 17:05:21 · 750 阅读 · 0 评论 -
将fastreid权重文件轻量化,并将其嵌入deepsort模型中,踩坑集锦
1.准备, 已经训练好的fast_reid配置文件和权重文件这是配置文件这是权重文件此时的权重文件共有282.10MB2.将配置文件转为onnx文件pip install onnxfast_reid代码(https://github.com/JDAI-CV/fast-reid/tree/master/tools/deploy)照着步骤走,先下载fastreid代码,cd 到tools/deploy文件夹下,然后终端运行命令:python onnx_export.py --conf原创 2021-11-10 19:12:03 · 2358 阅读 · 4 评论