Визуализация вывода промежуточных слоев cnn в pytorch

Question

Я пытаюсь визуализировать выходные данные промежуточных слоев сети VGG19 из модуля torchvision, в частности слоя conv4_2.Я извлек выход в тензор формы [1, 512, 50, 50].Но как мне визуализировать изображение с 512 каналами?

Answer 1

Визуализация объектов - очень сложная тема.
Если вы хотите получить наглядное представление о том, на что реагирует каждый фильтр (из 512) обученной сети, вы можете использовать такие методы, как these : распространение градиентов от вывода conv4_2 на входное изображение и измените изображение, чтобы максимизировать ответ функции. Вам нужно будет пройти через регуляризацию и т. Д., Чтобы получить плавные визуально приятные результаты.

В качестве альтернативы вы можете увидеть конкретные ответы каждого фильтра (из 512) на каждое местоположение (перекрывающиеся рецептивные поля). В этом случае у вас есть 512 различных изображений с интенсивностью 50 на 50, каждое из которых показывает карту ответов каждого нейрона на входное изображение.