1. 首页
  2. 编程语言
  3. Python
  4. YOLOV4-Tiny目标检测模型在TensorFlow 2中的高效实现

YOLOV4-Tiny目标检测模型在TensorFlow 2中的高效实现

上传者: 2023-12-08 21:36:19上传 ZIP文件 5.34MB 热度 57次

YOLOV4-Tiny目标检测模型在TensorFlow 2中的高效实现,充分支持多GPU训练,新增了对各个种类目标数量的精准计算以及热图(heatmap)功能。通过对损失函数的重新组织,确保了分类、目标检测和回归损失的比例适当。此外,该实现还具备支持步进(step)和余弦退火(cos)学习率调度的功能,提供了灵活的优化器选择,可选用Adam或SGD优化器。同时,该模型还实现了学习率根据批次大小自适应调整的功能,以及新增的图片裁剪功能,进一步提升模型的性能。

用户评论