Неверный размер при отображении изображения - PullRequest
Новая
       8

Неверный размер при отображении изображения

0 голосов
/ 06 декабря 2018

Я пытался найти скрытый слой VGG16 и отобразить карту объектов в Керасе.Я пытаюсь получить карту объектов block1_conv1 и отобразить ее.Но, к сожалению, я получаю сообщение об ошибке ниже: 

TypeError: Invalid dimensions for image data

Пожалуйста, найдите код ниже: 

from keras.applications.vgg16 import VGG16
from keras.preprocessing import image
from keras.applications.vgg16 import preprocess_input
from keras.models import Model
import matplotlib.pyplot as plt
import numpy as np
from keras import backend as K


img_path = "bombiliki.jpeg"
img = image.load_img (img_path, target_size=(224,224))
imgArr = image.img_to_array (img)
imgArr = np.expand_dims(imgArr, axis=0)
img = preprocess_input (imgArr)


model = VGG16(weights='imagenet', include_top=False)
layer_name = 'block1_conv1'

interMediateOutput = Model(inputs=model.inputs, outputs=model.get_layer(layer_name).output)
features = interMediateOutput.predict (img)
print ("Shape of the feature is ", features.shape)
pic = features[:,:,:,1]
print ("pic shape ", pic.shape)
data = np.asarray(pic)
print ("Data Dimension is ", data.ndim)

plt.imshow (pic)
plt.show()

Выход: 

('Shape of the feature is ', (1, 224, 224, 64))
('pic shape ', (1, 224, 224))
('Data Dimension is ', 3)
Traceback (most recent call last):
  File "vgg16.py", line 28, in <module>
    plt.imshow (pic)
  File "/home/navals/anaconda2/envs/musarni/lib/python2.7/site-packages/matplotlib/pyplot.py", line 3205, in imshow
    **kwargs)
  File "/home/navals/anaconda2/envs/musarni/lib/python2.7/site-packages/matplotlib/__init__.py", line 1855, in inner
    return func(ax, *args, **kwargs)
  File "/home/navals/anaconda2/envs/musarni/lib/python2.7/site-packages/matplotlib/axes/_axes.py", line 5487, in imshow
    im.set_data(X)
  File "/home/navals/anaconda2/envs/musarni/lib/python2.7/site-packages/matplotlib/image.py", line 653, in set_data
    raise TypeError("Invalid dimensions for image data")
TypeError: Invalid dimensions for image data

1 Ответ

0 голосов
/ 06 декабря 2018

Метод predict возвращает результат формы (n_samples, model_output_shape...).Поэтому, если вы дадите ему один образец, чтобы получить прогнозы для данного образца, вы должны сделать: 

pic = features[0]

И, в частности, в вашем примере, если вы хотите выбрать выход конкретного фильтра, вам нужночтобы указать его индекс в качестве четвертой оси: 

pic = features[0, :, :, desired_filter_index]

Они вы можете легко построить, что: 

plt.imshow(pic)
...