機械学習データ拡張で矩形のノイズをランダムで入れたい。

Question

プログラミング初心者で、今まではC言語くらいしかやってなかったので、pythonが全然わかりません。

このサイト(リンク内容)を参考にRandom Erasingを使い、画像内に矩形をいれ、データ拡張に使いたいと考えました。
別で使用した、ファイル内のすべての画像を読み込み、全部の画像に何かしらをするプログラムを利用して、組み合わせたのですがエラーが出てうまくいきません。

そもそもこのコードの書き方であっているのかもわかりません。pythonを1から勉強したいのはやまやまですが、使いながらなれないと時間がないので質問させていただきます。

さて、肝心のコードとエラーですが
コードは

import os
import glob
import cv2
import numpy as np

# glob.glob:条件に合致するすべてのファイルを取得
files = glob.glob('C:/Users/〇〇/*.jpg')

def random_erasing(img, p=1.0, sl=0.02, sh=0.4, r1=0.3, r2=3.3):
    target_img = img.copy()

    if p < np.random.rand():
        # RandomErasingを実行しない
        return target_img 

    H, W, _ = target_img.shape
    S = H * W

    while True:
        Se = np.random.uniform(sl, sh) * S # 画像に重畳する矩形の面積
        re = np.random.uniform(r1, r2) # 画像に重畳する矩形のアスペクト比

        He = int(np.sqrt(Se * re)) # 画像に重畳する矩形のHeight
        We = int(np.sqrt(Se / re)) # 画像に重畳する矩形のWidth

        xe = np.random.randint(0, W) # 画像に重畳する矩形のx座標
        ye = np.random.randint(0, H) # 画像に重畳する矩形のy座標

        if xe + We <= W and ye + He <= H:
            # 画像に重畳する矩形が画像からはみ出していなければbreak
            break

    mask = np.random.randint(0, 255, (He, We, 3)) # 矩形がを生成 矩形内の値はランダム値
    target_img[ye:ye + He, xe:xe + We, :] = mask # 画像に矩形を重畳

    return target_img

for file in files:
    img = cv2.imread(file)
    random_erasing(img)
    title, ext = os.path.splitext(file)
    
    cv2.imwrite(title + '_noise' + ext, img)

エラーは

  File "C:\Users\〇〇.spyder-py3\タイトル無し0.py", line 47, in <module>
    random_erasing(img)

  File "C:\Users\〇〇.spyder-py3\タイトル無し0.py", line 17, in random_erasing
    target_img = img.copy()

AttributeError: 'NoneType' object has no attribute 'copy'

と出ます。どうすれば解決するか教えてください。
よろしくお願いいたします。

＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃＃
追記です。エラーは出なくなったのですが、矩形が追加されません。どうすればよいでしょうか。

Answer 1

cv2.imreadでの画像ファイル読込に失敗してimgがNoneになっているようです。
ファイルパスに日本語を含むとcv2.imreadは正常に読み込めませんので、それが原因かと思われます。
参考：OpenCVでJPEG画像の読み取りができません。

Comments

[potapotapotato]

ファイルパスは問題なかったのですが、画像の名前に日本語が入っていました。
直したところ、エラーが起きず、実行できました。しかし、画像に矩形が追加されません。。。
どうすればよいでしょうか。

[potapotapotato]

自己解決しました。ありがとうございました！

Answer 2

cv2.imreadは読み込もうとしたファイルの画像フォーマットを判定できない場合、エラーを上げずにNoneを返します。

for file in files:
    img = cv2.imread(file)
    random_erasing(img)
    title, ext = os.path.splitext(file)

を以下のように書き換えて、エラーの出るファイルを特定し、その画像ファイルを調べて見てください。

for file in files:
    img = cv2.imread(file)
    if img == None:
        print('img is None for ', file)
    random_erasing(img)
    title, ext = os.path.splitext(file)

Comments

[potapotapotato]

ファイルパスは問題なかったのですが、画像の名前に日本語が入っていました。
直したところ、エラーが起きず、実行できました。しかし、画像に矩形が追加されません。。。
どうすればよいでしょうか。

[potapotapotato]

自己解決しました。ありがとうございました！

Answer 3

ファイルパスは問題なかったのですが、画像の名前に日本語が入っていました。
これを直すことにより、エラーはなくなりました。

また、矩形が表示されなかったのですが、for分の中の
random_erasing(img)
を
img = random_erasing(img)

に変更することにより解決しましたん

皆様。ありがとうございました。