def on_batch_end(self, batch, logs=None):
if batch % self.eval_per_batch != self.eval_per_batch - 1:
return
print('\nTest')
AP = local_eval(COCOeval, self.eval_model, self._image_size,
self.test_path, self.name_path, self.eval_n_samples,
self.verbose)
if AP > self._best_AP:
if self.save_path is None:
if self.verbose > 0:
print("AP improved from {:.2%} to {:.2%}".format(
self._best_AP, AP))
else:
save_path = self.save_path.format(mAP=AP)
if self.verbose > 0:
print(
"AP improved from {:.2%} to {:.2%}, saving model to {}"
.format(self._best_AP, AP, save_path))
if self.only_save_weight:
self.eval_model.save_weights(save_path)
else:
self.eval_model.save(save_path)
self._best_AP = AP
else:
if self.verbose > 0:
print("AP not improved from {:.2%}".format(self._best_AP))
def on_epoch_end(self, epoch, logs=None):
if epoch % self.sample_rate != self.sample_rate - 1:
return
mAP = local_eval(VOCeval, self.eval_model, self._image_size,
self.test_path, self.name_path, self.verbose)
if mAP > self._best_mAP:
if self.save_path is None:
if self.verbose > 0:
print("mAP improved from {:.2%} to {:.2%}".format(
self._best_mAP, mAP))
else:
save_path = self.save_path.format(mAP=mAP)
if self.verbose > 0:
print(
"mAP improved from {:.2%} to {:.2%}, saving model to {}"
.format(self._best_mAP, mAP, save_path))
self._best_mAP = mAP
self.eval_model.save_weights(save_path)
else:
if self.verbose > 0:
print("mAP not improved from {:.2%}".format(self._best_mAP))
def on_epoch_end(self, epoch, logs=None):
AP = local_eval(COCOEval, self.eval_model, self._image_size,
self.test_path, self.name_path, self.verbose)
if AP > self._best_AP:
if self.save_path is None:
if self.verbose > 0:
print("AP improved from {:.2%} to {:.2%}".format(
self._best_AP, AP))
else:
save_path = self.save_path.format(mAP=AP)
if self.verbose > 0:
print(
"AP improved from {:.2%} to {:.2%}, saving model to {}"
.format(self._best_AP, AP, save_path))
if self.only_save_weight:
self.eval_model.save_weights(save_path)
else:
self.eval_model.save(save_path)
self._best_AP = AP
else:
if self.verbose > 0:
print("AP not improved from {:.2%}".format(self._best_AP))