logo标签列表

在TensorFlow目标检测API中检测时期结束

tensorflow
3

如何在TF Object Detection API中检测一个epoch的结束（即完成对数据集的一次完整遍历）？这可能有助于进行一些簿记或一些内部处理（例如重置一些权重）在自定义检测模型中。

1个回答
0

你可能想要实现 tf.estimator.SessionRunHook

为此，您需要在tf.estimator.TrainSpec的model_lib.py中添加hook参数，或者创建自己的训练文件并在将其传递给tf.estimator.train_and_evaluate之前覆盖train_spec。

使用ProfilerHook添加到Tensorflow Object Detection API的示例： （应该类似于SessionRunHook）

config = tf.estimator.RunConfig(model_dir=model_dir, save_checkpoints_steps=save_checkpoints_steps,
                            save_checkpoints_secs=save_checkpoints_secs, keep_checkpoint_max=keep_checkpoint_max,
                            log_step_count_steps=log_step_count_steps)

train_and_eval_dict = model_lib.create_estimator_and_inputs(
      run_config=config,
      hparams=model_hparams.create_hparams(hparams_overrides),
      pipeline_config_path=pipeline_config_path,
      config_override = cfg_override,
      train_steps=num_train_steps,
      sample_1_of_n_eval_examples=sample_1_of_n_eval_examples,
      sample_1_of_n_eval_on_train_examples=sample_1_of_n_eval_on_train_examples,
      save_final_config=save_final_config)

estimator = train_and_eval_dict['estimator']
train_input_fn = train_and_eval_dict['train_input_fn']
eval_input_fns = train_and_eval_dict['eval_input_fns']
eval_on_train_input_fn = train_and_eval_dict['eval_on_train_input_fn']
predict_input_fn = train_and_eval_dict['predict_input_fn']
train_steps = train_and_eval_dict['train_steps']

train_spec, eval_specs = model_lib.create_train_and_eval_specs(
  train_input_fn,
  eval_input_fns,
  eval_on_train_input_fn,
  predict_input_fn,
  train_steps,
  eval_on_train_data=False)

profile_hook = tf.train.ProfilerHook(save_steps=profiler_save_step, save_secs=None, output_dir=profiler_output_dir, 
                                     show_dataflow=True, show_memory=True)

train_spec = tf.estimator.TrainSpec(input_fn=train_input_fn
                                    ,max_steps=train_steps
                                    ,hooks=[profile_hook])

tf.estimator.train_and_evaluate(estimator, train_spec, eval_specs[0])
网页内容由stack overflow 提供, 点击上面的
可以查看英文原文，
原文链接