模型转换
模型转换用于将YOLOv5 Torch模型转换为TensorRT格式,可以极大提升检测性能。
功能入口
- 功能入口:工具箱->机器学习->视频分析->模型转换。
参数说明
- 训练数据路径:选择选择通过图片样本管理生成的训练数据所在的文件夹。
- 模型:选择需要转换的模型。
- 训练图片大小:设置用于模型训练的图片大小,需要与模型训练的图片大小一致。
- 视频宽:设置视频的宽度,视频宽必须是转换后需要检测视频的宽度。
- 视频高:设置视频的高度,视频高必须是转换后需要检测视频的高度。
- 转换后的模型只能用于检测所设置的视频宽高比例的视频,如设置宽高为1920*1080,则该模型只能用于1920*1080、1280*720等视频。
注意事项:
|