diff --git a/train.py b/train.py index bbef15c846a0558c9c3b658817ec4829a93b6dbe..f9b234f33a2fce150b8362c443712a57297afcad 100644 --- a/train.py +++ b/train.py @@ -207,9 +207,10 @@ if __name__ == "__main__": #------------------------------------------------------# input_shape = (416,416) #------------------------------------------------------# - # 是否对损失进行归一化 + # 是否对损失进行归一化,用于改变loss的大小 + # 用于决定计算最终loss是除上batch_size还是除上正样本数量 #------------------------------------------------------# - normalize = True + normalize = False #----------------------------------------------------# # 获取classes和anchor