1. 首页
  2. 人工智能
  3. 深度学习
  4. 基于yolov5的深度学习目标检测程序及环境配置教程

基于yolov5的深度学习目标检测程序及环境配置教程

上传者: 2023-07-01 17:20:55上传 ZIP文件 926.92MB 热度 12次

基于yolov5算法的深度学习目标检测程序,可通过该教程完成配置环境并直接开始训练模型。YOLOV5是基于pytorch实现的You Only Look Once目标检测模型(Ultralytics版本为v5.0),支持多GPU训练,新增了各个种类目标数量计算和heatmap功能。此外,程序还针对正样本匹配过程进行了优化,提升了运算速度,并引入了EMA算法以改善模型效果。2022年02月创建了该仓库,支持不同尺寸模型的训练(包括s、m、l和x版本的yolov5),支持step和cos学习率下降法,可选择adam或sgd优化器,还能根据batch_size自适应调整学习率,并新增了图片裁剪功能。

用户评论