Speichern Sie tf.summary.image mit Estimator API

Question

im Moment versuche ich, ein wenig vertrauter mit der TF Estimator API zu bekommen. Ich arbeite/lerne mit dem Beispiel aus dieser blog entry.

Jetzt habe ich das Problem, dass ich eine einfache Bildzusammenfassung (tf.summary.image) nicht speichern kann.

Ich habe eine SummarySaverHook im experiment_fn und den Haken zu den eval_hooks Parameter des Experiments Konstruktor hinzu:

def experiment_fn(run_config, params): 
    ... 
    summary_hook = tf.train.SummarySaverHook(
     100, 
     output_dir='/summeries', 
     summary_op=tf.summary.merge_all() 
    ) 
    ... 
    experiment = tf.contrib.learn.Experiment(
     estimator=estimator, # Estimator 
     train_input_fn=train_input_fn, # First-class function 
     eval_input_fn=eval_input_fn, # First-class function 
     train_steps=params.train_steps, # Minibatch steps 
     min_eval_frequency=params.min_eval_frequency, # Eval frequency 
     train_monitors=[train_input_hook], # Hooks for training 
     eval_hooks=[eval_input_hook, summary_hook], # Hooks for evaluation 
     eval_steps=None # Use evaluation feeder until its empty 
    ) 

Was mir nicht klar ist, wo ich das hinzufügen müssen tf.summary.image() Funktion, damit das Bild in der Bildregisterkarte des Tensorboards angezeigt wird.

Vielen Dank im Voraus.

Answer 1

Sie sollten keinen Haken hinzufügen müssen. Fügen Sie einfach den tf.summary.image Anruf irgendwo in Ihrem model_fn/input_fn hinzu und der Schätzer sollte automatisch einen Zusammenfassungs-Hook für alle erstellten Zusammenfassungen hinzufügen.