return {"feature_extractor": self._feature_extractor.get_base_model()}
if fine_tune_checkpoint_type == "detection":
return {"feature_extractor": self._feature_extractor.get_model()}
else:
raise ValueError("Unknown fine tune checkpoint type - {}".format(
fine_tune_checkpoint_type))
After Change
def restore_map(self,
fine_tune_checkpoint_type="detection",
load_all_detection_checkpoint_vars=False):
raise RuntimeError("CenterNetMetaArch not supported under TF1.x.")
def restore_from_objects(self, fine_tune_checkpoint_type="detection"):
Returns a map of Trackable objects to load from a foreign checkpoint.