ディレクトリ内ファイルの一括処理について

Question

ディレクトリ内の画像処理について、複数の画像を一括で赤（255,0,0）→（1,0,0）にするコードで悩んでいます。
自分ないに書いてみたのですが、TypeError: string indices must be integersエラーがでてうまくいきません。

import cv2
import os

設定(org)
img_dir = '入力ディレクトリ ’
out_dir = '出力ディレクトリ'
os.makedirs(out_dir, exist_ok=True)

WIDTH=1000
HEIGHT=500

path="./パス名"
files=os.listdir(path)

print(files)

for file in files:
for x in range(HEIGHT):
for y in range(WIDTH):
r,g,b = file[x, y]
if (r, g, b) == (255, 0, 0):
file[x,y]=1,0,0
else:
file[x,y]=0,0,0

print(files)

上記のコードでおかしい点を教えていただきたいです。
よろしくお願いします。

Answer 1

以下の問題点があるように思います。

1. OpenCV はチャンネルの並び順が BGR なので、赤は (255, 0, 0) ではなく、(0, 0, 255)
2. numpy 配列に対して、for 文で1ピクセルずつ見ていくのは非効率

→ numpy の indexing を使う。

サンプルコード

from pathlib import Path

import cv2
import numpy as np

input_dir = Path('input')
output_dir = Path('output')
output_dir.mkdir(exist_ok=True)

before = (0, 0, 255)
after = (0, 0, 1)

for img_path in input_dir.iterdir():
    # 画像を読み込む。
    img = cv2.imread(str(img_path))

    # (0, 0, 255) を (1, 0, 0) に置換する。
    idx_to_replace = (img == before).all(axis=2)
    img[idx_to_replace] = after
    img[~idx_to_replace] = (0, 0, 0)

    # 保存する。
    save_path = output_dir / img_path.name
    cv2.imwrite(str(save_path), img)
    print('{} ==> {}'.format(img_path, save_path))

Comments

[退会済みユーザー]

ご丁寧にありがとうございます。
記載し忘れていましたが、（0,0,255）以外はすべて(0,0,0)で処理したいのですが、上記のプログラムに加える方法を教えていただけないでしょうか。
お手数おかけします。

[tiitoi]

回答を修正しました。
なお、(0,0,0)と(0, 0, 1)は近い色なので、画像で見ると真っ黒になると思います。

[退会済みユーザー]

ありがとうございます！
無事解決しました。

Answer 2

fileは文字列です。文字列に対してr,g,b = file[x, y]という操作ができないので提示エラーが発生しています。
img = cv2.imread(file)の結果に対して処理をしましょう。

Comments

[退会済みユーザー]

ありがとうございます。
いろんな方法で試したいと思います。