Как отобразить количество эпох в API обнаружения объектов тензорного потока с помощью Faster Rcnn?

Я использую API обнаружения объектов Tensorflow. Я понял, читая файл fast_rcnn_inception_v2_pets.config, что num_steps означает общее количество шагов, а не эпох. Но тогда какой смысл указывать batch_size ?? Допустим, у меня есть 500 изображений в моих обучающих данных, и я установил размер пакета = 5 и num_steps = 20k. Значит ли это, что количество эпох равно 200 ??

Когда я запускаю model_main.py, он показывает только потерю global_steps. Итак, если эти глобальные шаги не являются эпохами, то как мне изменить код, чтобы отображать потерю поездов и потерю val после каждого шага, а также после каждой эпохи?


python object-detection-api tensorflow
person Ps98    schedule 20.06.2019    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Итак, вы правы в своем предположении, что у вас 200 эпох.

У меня была аналогичная проблема с не показом проигрыша. Мое решение заключалось в том, чтобы перейти к файлу model_main.py, а затем вставить

tf.logging.set_verbosity (tf.logging.INFO)

после импорта. затем он показывает потери после каждых 100 шагов. вы можете изменить set_verbosity, если хотите, чтобы он был после каждой эпохи;)

person Tonto5000    schedule 12.07.2019
comment
На самом деле я хотел знать, как отображать потери train и val после каждого шага. Также, поскольку я установил оценку после 100 шагов, поэтому она не будет оцениваться после каждого шага. Таким образом, одно из решений - установить оценку после каждого шага, и для этого необходимо указать правильный размер пакета в конфигурации eval. У вас есть лучшее решение, так как это сильно замедлит мои тренировки. - person Ps98; 14.07.2019