模型转换

模型转换用于将YOLOv5 Torch模型转换为TensorRT格式,可以极大提升检测性能。

功能入口

  • 功能入口工具箱->机器学习->视频分析->模型转换

参数说明

  • 训练数据路径:选择选择通过图片样本管理生成的训练数据所在的文件夹。
  • 模型:选择需要转换的模型。
  • 训练图片大小:设置用于模型训练的图片大小,需要与模型训练的图片大小一致。
  • 视频宽:设置视频的宽度,视频宽必须是转换后需要检测视频的宽度。
  • 视频高:设置视频的高度,视频高必须是转换后需要检测视频的高度。
  • 注意事项
    • 转换后的模型只能用于检测所设置的视频宽高比例的视频,如设置宽高为1920*1080,则该模型只能用于1920*1080、1280*720等视频。

相关内容

视频分析环境配置

模型训练

时间识别