diff --git a/train.py b/train.py index f9b234f33a2fce150b8362c443712a57297afcad..573beba8e39bfba9fdf275a6a2a82d73d7272cf6 100644 --- a/train.py +++ b/train.py @@ -224,11 +224,12 @@ if __name__ == "__main__": num_anchors = len(anchors) #------------------------------------------------------# # Yolov4的tricks应用 - # mosaic 马赛克数据增强 True or False + # mosaic 马赛克数据增强 True or False + # 实际测试时mosaic数据增强并不稳定,所以默认为False # Cosine_scheduler 余弦退火学习率 True or False # label_smoothing 标签平滑 0.01以下一般 如0.01、0.005 #------------------------------------------------------# - mosaic = True + mosaic = False Cosine_scheduler = False label_smoothing = 0