基于yolov5的深度学习目标检测程序及环境配置教程
基于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自适应调整学习率,并新增了图片裁剪功能。
用户评论