
YOLOV7-OBB:基于PyTorch的You Only Look Once OBB旋转目标检测模型实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:YOLOV7-OBB是基于PyTorch框架开发的一种先进的旋转边界框(OBB)目标检测模型,继承了YOLO系列高效准确的特点。该模型能够精确识别图像中各类物体的旋转角度和位置,尤其适用于复杂场景下的小目标及倾斜目标检测任务。
在YOLOv7的基础上使用KLD损失函数改进为旋转目标检测模型yolov7-obb;支持step、cos学习率下降法、优化器选择包括adam和sgd、根据batch_size自适应调整学习率、新增图片裁剪功能、多GPU训练支持、计算各类别目标数量统计,同时支持heatmap以及EMA。
全部评论 (0)
还没有任何评论哟~


