Keras Cropping2Dカラーチャネル

Question

を変更し、私は次のコードを使用して2Dクロッピングkerasの効果を視覚化しようとした：ここ

from keras import backend as K 
from keras.layers.convolutional import Cropping2D 
from keras.models import Sequential 
# with a Sequential model 
model = Sequential() 
model.add(Cropping2D(cropping=((22, 0), (0, 0)), input_shape=(160, 320, 3))) 
cropping_output = K.function([model.layers[0].input], 
            [model.layers[0].output]) 
cropped_image = cropping_output([image[None,...]])[0] 
compare_images(image, 
       cropped_image.reshape(cropped_image.shape[1:])) 

はプロット関数である：

def compare_images(left_image, right_image):  
    print(image.shape) 
    f, (ax1, ax2) = plt.subplots(1, 2, figsize=(24, 9)) 
    f.tight_layout() 
    ax1.imshow(left_image) 
    ax1.set_title('Shape '+ str(left_image.shape), 
        fontsize=50) 
    ax2.imshow(right_image) 
    ax2.set_title('Shape '+ str(right_image.shape) 
        , fontsize=50) 
    plt.show() 

結果は

あります

明らかに、カラーチャネルが変更されているd。しかし、なぜ？私のコードにエラーがありますか、それともケラスのバグでしょうか？

Answer 1

これはKerasのバグではありません。テンソルは通常float32タイプなので、出力が評価されるときにはfloat32タイプもあります。表示する前に、画像データをuint8に変換する必要があります。

ax2.imshow(np.uint8(right_image)) 

compare_imagesは、画像を正しく表示する必要があります。