ValueError: could not broadcast input array from shape (75,75,3) into shape (75,75) について

Question

https://qiita.com/kkk3H/items/c3eb0d868170b29b0b87

上記の記事を参考にして画像データとラベルデータを作ろうとしたのですが
ValueError: could not broadcast input array from shape (75,75,3) into shape (75,75)
というエラーが出てしまいます。画像のデータがいけないのでしょうか？　よろしくお願いします。

画像データはグレースケール画像ではなくRGB画像です。
もしかしたらそこがいけないのかもしれません。。。

Python 3.6.4
TensorFlow 1.3.0
NumPy 1.13.3

import numpy
from PIL import Image
import glob
from tensorflow.python.framework import dtypes
from numpy.random import *
import os
numpy.random.seed(0)


if __name__ == '__main__':

    directory_path = '/Users/lab/data_set'
    file_format = 'png'
    shuffle = True

    A_path = os.path.join(directory_path, 'A')
    B_path = os.path.join(directory_path, 'B')
    C_path = os.path.join(directory_path, 'C')

    all_images = []
    all_labels = []

    label = 0

    for path in [A_path, A_path, A_path]:
        img_files = glob.glob(path + '/*.' + file_format)

        for img_file in img_files:
            img = numpy.array(Image.open(img_file))
            all_images.append(img)
            all_labels.append(label)

        label += 1

    all_images = numpy.array(all_images)
    all_labels = numpy.array(all_labels)

    indices = numpy.arange(len(all_images))

    if shuffle:
        numpy.random.shuffle(indices)

    all_images = all_images[indices]
    all_labels = all_labels[indices]

    #print (len(all_labels))
    numpy.save('./imgs.npy', all_images)
    numpy.save('./labels.npy', all_labels)

Answer 1

お察しの通り、以下の部分が問題でしょうね。
グレースケールの場合、1ピクセルにあるデータは１つとなってますが、RGBの場合１ピクセルには3つのデータが存在します。

画像データはグレースケール画像ではなくRGB画像です。

もしかしたらそこがいけないのかもしれません。。。

# 例）3*3pxの画像データを想定した時のデータはこの様なイメージです。

# グレーの場合
[[2,5,6],
 [5,4,2],
 ......
 [8,4,1]]

# RGBの場合
[[[2,5,4],[5,6,7],[8,9,2],
 [[5,4,2],[6,8,4],[1,2,4]],
 ......
 [[5,1,4],[2,6,8],[4,1,5]]]

では、どうすればいいかですが、、、例えばOpenCVというライブラリを使用して色の変換をしてしまう等すればいいと思います。ただ、最後に変数の型を正す必要が出てくるかもしれないので、そこはご自身でやってみてください。

import cv2

img = cv2.imread('パス')
gray_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

Opencvのドキュメント

Comments

[ruuruusann24]

回答ありがとうございます！！ わかってきたのですが、うまく出来ません。。。どのようにすれば良いか具体的に教えていただけないでしょうか。。。時間があまりない状況でして。。。すみません。。。

[Yuki_S]

これで動きませんかね？試してないので自信はないですが・・・

　for img_file in img_files:
　　　　img = cv2.imread(img_file)
　　　　gray_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
　　　　all_images.append(gray_img)
　　　　all_labels.append(label)

[ruuruusann24]

遅くなって大変申し訳ありません！おかげさまで助かりました！ありがとうございます！またわからないところがあったらよろしくお願いします！

[Yuki_S]

うまくいったのであればよかったです！お役にたてそうなことあればぜひともお願いします。