Tensorflow object detection API not displaying global steps

前端 未结 3 956
无人及你
无人及你 2021-02-20 02:38

I am new here. I recently started working with object detection and decided to use the Tensorflow object detection API. But, when I start training the model, it does not display

相关标签:
3条回答
  • 2021-02-20 03:25

    As Thommy257 suggested, adding tf.logging.set_verbosity(tf.logging.INFO) after the import section of model_main.py prints the summary after every 100 steps by default.

    Further, to specify the frequency of the summary, change

    config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir)

    to

    config = tf.estimator.RunConfig(model_dir=FLAGS.model_dir, log_step_count_steps=k)

    where it will print after every k steps.

    0 讨论(0)
  • 2021-02-20 03:26

    Add tf.logging.set_verbosity(tf.logging.INFO) after the import section of the model_main.py script. It will display a summary after every 100th step.

    0 讨论(0)
  • 2021-02-20 03:33

    Regarding the recent change to model_main , the previous version is available at the folder "legacy". I use train.py and eval.py from this legacy folder with the same functionality as before.

    0 讨论(0)
提交回复
热议问题