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

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

python
1import numpy
2from PIL import Image
3import glob
4from tensorflow.python.framework import dtypes
5from numpy.random import *
6import os
7numpy.random.seed(0)
8
9
10if __name__ == '__main__':
11
12    directory_path = '/Users/lab/data_set'
13    file_format = 'png'
14    shuffle = True
15
16    A_path = os.path.join(directory_path, 'A')
17    B_path = os.path.join(directory_path, 'B')
18    C_path = os.path.join(directory_path, 'C')
19
20    all_images = []
21    all_labels = []
22
23    label = 0
24
25    for path in [A_path, A_path, A_path]:
26        img_files = glob.glob(path + '/*.' + file_format)
27
28        for img_file in img_files:
29            img = numpy.array(Image.open(img_file))
30            all_images.append(img)
31            all_labels.append(label)
32
33        label += 1
34
35    all_images = numpy.array(all_images)
36    all_labels = numpy.array(all_labels)
37
38    indices = numpy.arange(len(all_images))
39
40    if shuffle:
41        numpy.random.shuffle(indices)
42
43    all_images = all_images[indices]
44    all_labels = all_labels[indices]
45
46    #print (len(all_labels))
47    numpy.save('./imgs.npy', all_images)
48    numpy.save('./labels.npy', all_labels)
49

回答1件

ベストアンサー

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

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

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

Python
1# 例）3*3pxの画像データを想定した時のデータはこの様なイメージです。
2
3# グレーの場合
4[[2,5,6],
5 [5,4,2],
6 ......
7 [8,4,1]]
8
9# RGBの場合
10[[[2,5,4],[5,6,7],[8,9,2],
11 [[5,4,2],[6,8,4],[1,2,4]],
12 ......
13 [[5,1,4],[2,6,8],[4,1,5]]]
14

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

Python
1import cv2
2
3img = cv2.imread('パス')
4gray_img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

Opencvのドキュメント

投稿2018/01/22 04:47

Yuki_S

総合スコア356

回答へのコメント

ruuruusann24

2018/01/22 06:05

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

Yuki_S

2018/01/22 06:12 編集

これで動きませんかね？試してないので自信はないですが・・・　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

2018/01/30 16:53

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

Yuki_S

2018/01/31 00:29

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