Tensorflow inception-V3 Re-Train несколько слоев

Question

Я успешно прошел переподготовку начального уровня V3, используя Python2.7 api для своих собственных 100 классов, и он дает достойные результаты, но не исключительно хорошо.

У меня также есть код для переподготовки всей сети с нуля, как указано here (google code), но это ресурс и интенсивность времени, и у меня 400 000 изображений, поэтому не знаю, какая будет точность после тренировки.

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

Я пытался найти много, но ничего не нашел. Возможно, что я хочу делать? и мне нужна помощь в этом.

Answer 1

Оптимизаторы смотрят на список «обучаемых переменных». Ссылка на этот массив доступна с использованием tf.get_collection_ref(tf.GraphKeys.TRAINABLE_VARIABLES). Вы должны иметь возможность изменить это. Версия для чтения - tf.trainable_variables.

Обратите внимание, что шаг вперед (вывод) всегда должен выполняться, поэтому вам придется заплатить эту стоимость. Если вы этого не хотите, проще всего запустить вывод, сохранить вывод в tf-записи, а затем обучить из этих предварительно обработанных образцов данных.