機械学習について 画像の水増し

Question

人物判別のために画像を水増ししたいのですがエラーが出てしまいます

import os
import numpy as np
import matplotlib.pyplot as plt
import cv2
from os import listdir

def  make_image(input_img):
    img_size = input_img.shape
    filter_one = np.ones((3, 3))
    fake_method_array = np.array([
       lambda x: cv2.threshold(x, 100, 255, cv2.THRESH_TOZERO)[1],
        lambda x: cv2.GaussianBlur(x, (5, 5), 0),
        lambda x: cv2.resize(cv2.resize( x, (img_size[1] // 5, img_size[0] // 5)),(img_size[1], img_size[0])),
        lambda x: cv2.erode(x, filter_one),
        lambda x: cv2.flip(x, 1),
    ])
    images = []
    for method in fake_method_array:
        faked_img = method(input_img)
        images.append(faked_img)
    return images
  
path = "人物判別/downloads/matsumoto hitoshi/"
imagesList = listdir(path)
i=1
for file in imagesList:
    target_img = cv2.imread(path+file)
    fake_images = make_image(target_img)
    if not os.path.exists("matsumoto_images"):
        os.mkdir("matsumoto_images")
    for number, img in enumerate(fake_images):
        cv2.imwrite("matsumoto_images" + str(i)+str(number) + ".jpg" ,img) 
    i=i+1

Traceback (most recent call last):
  File "c:/Users/my174/programming/python/machine learning/5_04.py", line 28, in <module>
    fake_images = make_image(target_img)
  File "c:/Users/my174/programming/python/machine learning/5_04.py", line 8, in make_image
    img_size = input_img.shape
AttributeError: 'NoneType' object has no attribute 'shape'

環境はpython3.8.3
vscode
ディレクトリ構成は　1人物判別　2downloads 3matsumoto hitoshi　
　　　　　　　　　　　      　　　　　　　3.5_04.py
です。

Comments

[meg_]

エラーが出ているのは「img_size = input_img.shape」ですか？

[law]

すみません。初心者なのでどこがエラーなのかわかりません。

Answer 1

megさんとLuKeさん本当にありがとうございました！
すべてローマ字に変更して/をいれたら無事に解決しました。

Answer 2

二回回答してしまいました。すみません。
これからもよろしくお願いします。

Answer 3

###原因
パスが原因だと思われます。(以下はその説明です。面倒であれば飛ばしても大丈夫です)
ソースコードでshapeが使われているのはmake_image関数の定義の場所です。そして、そのshapeはmake_image関数の仮引数input_imgのものです。
また、input_imgには引数を受け取ってからなんの操作もしていないことから、このエラーはmake_image関数使用時に起こっている、ということになります。
ソースコードをみると、make_image関数使用時の実引数はtarget_imgで、これは

    target_img = cv2.imread(path+file)

この行で定義されています。
原因はこの行のcv2.imreadの実引数である

path+file

にあります。
変数pathには「"人物判別\downloads\松本人志"」が代入されています。
そして、変数fileには、例えば「"001.png"」のような文字列が代入されています。
この2つをつなげると「"人物判別\downloads\松本人志001.png"」となり、この様なファイルが存在しないのでcv2.imreadはNoneTypeを返しています。

解決方法

これを解決するには、pathの値を
"人物判別\downloads\松本人志"
ではなく、
"人物判別\downloads\松本人志"
のようにすればよいです。
※文字列でバックスラッシュ()を使うとエスケープシーケンスと間違われる可能性があるので、普通のスラッシュ(/)を使うほうが好ましいです。

Comments

[law]

すみません。変更して実行しましたが変化なしで同じエラーが出ます。

[Luke02561]

では、
target_img = cv2.imread(path+file)
の前にtarget_imgにどんなデータが入っているのか確認してみてください

[law]

target_imgというフォルダが必要なのでしょうか？
画像データと上記のプログラムしかデータは入っていません。

[Luke02561]

すみません、target_imgという変数にどの様な値が入っているか、ということです。
print関数でだいたい見られます。

[law]

print(target_img)で実行したところNoneという結果が出ました。

[Luke02561]

では、imagesListの内容をすべて確認してみてください。
画像ファイル以外は何も入っていませんか？

[law]

画像だけです

[Luke02561]

となるとパスが間違っていることしか考えられませんが...
path+file
の値はどうなっていますか？

[law]

人物判別/downloads/matsumoto hitoshi/000001.jpgと表示されました。

[Luke02561]

では、pathを相対パスではなく、絶対パスで指定してみてください。
参考に：https://qiita.com/right1121/items/af0c3555b75fd3e51bdf

もう一つ考えられるのが、実行ディレクトリです。
例えば、
A
|
|--B
| 　 |--python.py
| 　|--img.png

といった構造のディレクトリがあり、コンソール上で、Aというフォルダから実行した場合、img.pngは正しく読み込まれません。

[law]

パスでも変わりませんでした。ディレクトリ構成はどのようにしたらよいのでしょうか？

[Luke02561]

実行するファイルと、画像のあるフォルダをおなじフォルダにいれるといいです。
また、実行する場所も、
Name@hoge:~/(実行するフォルダのあるファイル)$
この様な場所で実行してください。
実行するフォルダのあるファイルとは、上記の構造の例だと、A/Bのことです

[law]

downloadsに.pyファイルをいれて実行しましたが変化なしです。

[Luke02561]

実行するプログラムは、「人物判別」のあるディレクトリと同じところでお願いします。
ちなみにpathを
"./人物判別/downloads/matsumoto hitoshi/"
とするとどうでしょうか

[law]

すみません。変わりませんでした。

Answer 4

OSが不明ではありますが、日本語を含むパスの問題かと思われます。

Python OpenCV の cv2.imread 及び cv2.imwrite で日本語を含むファイルパスを取り扱う際の問題への対処について

ディレクトリ名を変更するのが早いかとは思いますが。

Comments

[law]

すべてローマ字表記にしましたが同じerrorが表示されます。

[meg_]

「人物判別」が日本語ですね。（Luke02561さんご指摘の"/"の不足と合わせてこちらを変更したら解決しませんか？）