質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.58%

python+opencvで、白黒画像とカラー画像を分類したい

解決済

回答 2

投稿

  • 評価
  • クリップ 0
  • VIEW 2,712

anpuru

score 13

 実現したいこと

pythonとopencvを利用して、読み込んだ画像が白黒画像とカラー画像とに分類したい。

 困っていること

opencvの関数を使って、画像の色の判定を行えばいいと思うのですが、調べたのですが具体的な方法が分かりませんでした。
どなたかご教授いただけないでしょうか。よろしくお願いいたします。

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

質問への追記・修正、ベストアンサー選択の依頼

  • t_obara

    2018/10/19 16:00

    元の画像フォーマットは決まっているのでしょうか?

    キャンセル

  • anpuru

    2018/10/19 16:03

    画像フォーマットは、jpegです。

    キャンセル

回答 2

checkベストアンサー

0

PIL を使った例
もし PIL が入っていなければ、以下で入れてください。

pip install pillow
import glob
import os

from PIL import Image

input_dirpath = 'input'  # 入力ディレクトリ
output_dirpath = 'output'  # 出力ディレクトリ

# ファイルパスをピックアップ
types = ('*.png', '*.gif', '*.jpg')
img_paths = []
for t in types:
    img_paths.extend(glob.glob(os.path.join(input_dirpath, t)))

# リサイズする。
for path in img_paths:
    img = Image.open(path)
    if img is None:
        print('failed to loading image {}.'.format(path))
        continue
    # 分類する。
    if img.mode == 'RGBA':
        dirpath = os.path.join(output_dirpath, 'RGBA')
    elif img.mode == 'RGB':
        dirpath = os.path.join(output_dirpath, 'RGB')
    elif img.mode == 'L':
        dirpath = os.path.join(output_dirpath, 'L')
    else:
        print('image {}: unsupported imaga type: {}'.format(path, img.mode))
        continue
    os.makedirs(dirpath, exist_ok=True)
    print('image {}: {}'.format(path, img.mode))

    save_path = os.path.join(dirpath, os.path.basename(path))
    img.save(save_path)

 追記

白と黒のみで構成される画像は「2値画像」といいます。

import glob
import os
import cv2

input_dirpath = 'tet'  # 入力ディレクトリ
output_dirpath = 'output'  # 出力ディレクトリ

# ファイルパスをピックアップ
types = ('*.png', '*.gif', '*.jpg')
img_paths = []
for t in types:
    img_paths.extend(glob.glob(os.path.join(input_dirpath, t)))

# リサイズする。
for path in img_paths:
    img = cv2.imread(path)
    if img is None:
        print('failed to loading image {}.'.format(path))
        continue

    # 分類する。
    colors = np.unique(img.reshape(-1, img.shape[2]), axis=0)
    if len(colors) <= 2 and np.all(np.logical_or(np.all(colors == [0, 0, 0], axis=1),
                                                 np.all(colors == [255, 255, 255], axis=1))):
       dirpath = os.path.join(output_dirpath, 'binary')
    else:
       dirpath = os.path.join(output_dirpath, 'color')

    os.makedirs(dirpath, exist_ok=True)
    save_path = os.path.join(dirpath, os.path.basename(path))
    cv2.imwrite(save_path, img)

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2018/10/19 22:16

    ご回答ありがとうございます。
    ご丁寧にありがとうございます。

    完全に白と黒ではなくて、白っぽい色と黒っぽい色なのですが、そのような指定は可能なのでしょうか

    キャンセル

  • 2018/10/20 01:52 編集

    白っぽい、黒っぽいというのは質問者さんの主観によりますので、方針だけ書きますね。
    白系の色は (255, 255, 255) に近い色、黒系の色は (0, 0, 0) に近い色です。
    なので、各画素を調べて、白系の色、黒系の色、それ以外の色の画素数をそれぞれ数えて、画像全体に占める白系の色、黒系の色の画素が多ければ、白黒画像、そうでなければ、カラー画像として判断すればよいかと思います。

    キャンセル

  • 2018/10/20 10:38

    参考にして調べてみます。ご回答ありがとうございました。

    キャンセル

0

こちらでいかがでしょうか。

>>> import cv2
>>> import numpy as np

>>> img = cv2.imread('messi5.jpg')
>>> print (img.shape)
(342, 548, 3)


上記shapeですが、グレースケールの場合、行数と列数しか含まないとのことです。

投稿

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.58%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る