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