Как правильно выбрать модель передачи в стиле Google?

Question

, пожалуйста, помогите!

Я начал выполнять задачу, используя шаги Google оттуда: https://www.tensorflow.org/tutorials/generative/style_transfer

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

Я попытался открыть URL из кода с точно используемой моделью, и я вижу хранилище этой модели https://tfhub.dev/google

но моделей очень много. Как я могу получить правильный?

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

Пожалуйста, помогите мне с решением - как я могу получить или Как правильно найти подходящую модель для этой задачи?

Answer 1

В настоящее время для передачи стиля существует одна модель https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2. Существует множество обобщенных c моделей изображений, полезных для других задач на основе изображений, таких как классификация. Если вы тренируете свою собственную модель передачи стиля, вы можете поделиться ею с сообществом, опубликовав ее на TensorFlow Hub.