iou_score не улучшился - PullRequest
Новая
       6

iou_score не улучшился

0 голосов
/ 11 апреля 2020

В моем коде, хотя iou_score , кажется, улучшается, рано останавливается после некоторых эпох. Может ли кто-нибудь просветить меня ?? enter image description here

здесь часть кода: 

import segmentation_models as sm

dice_loss = sm.losses.DiceLoss()
# focal_loss = sm.losses.BinaryFocalLoss() 
# total_loss = dice_loss + (1 * focal_loss)
metrics = [sm.metrics.IOUScore(threshold=0.5), sm.metrics.FScore(threshold=0.5)]

from keras import optimizers
adam=optimizers.Adam(lr=0.0001, beta_1=0.9, beta_2=0.999, amsgrad=False)

model.compile(loss=dice_loss,optimizer='adam',metrics=metrics)



earlystopper = EarlyStopping(monitor='iou_score',patience=5, verbose=1,mode='auto')
checkpointer = ModelCheckpoint(filepath=r'C:\Users\User\Desktop\SeeFar2\Folder with Results\Semantic_model2.h5',
                               monitor='iou_score',verbose=1, save_best_only='True', save_weights_only='False',mode='auto')
...