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))