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

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

ただいまの
回答率

87.48%

画像にどんな色が使われているのかの調査

受付中

回答 0

投稿

  • 評価
  • クリップ 0
  • VIEW 531

score 5

実現したいこと

入力された画像をk-means法で減色し画像に使われている色を抽出したいです。
例えば下の画像なら(48, 153, 233), (65, 192, 134), (114, 98, 218), (192, 188, 65)を出力したいです。
イメージ説明

発生している問題

入力する画像が上の画像なら正常に動作するのですが画像によってはエラーが発生しte

該当のソースコード

import numpy as np
import cv2
from matplotlib import pyplot as plt

img = cv2.imread('tree.jpg')

# マスキングして白の背景を取り除く
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
threshold_value = 200
ret, threshold_img = cv2.threshold(gray, threshold_value, 255, cv2.THRESH_BINARY)
threshold_img = ~threshold_img
masked_img = cv2.cvtColor(threshold_img, cv2.COLOR_GRAY2BGR)
img = masked_img & img


# メディアンフィルタで端をなめらかにする。
threshold_img = cv2.medianBlur(threshold_img, 7)

# 輪郭抽出を実行する。
contours, hierarchy = cv2.findContours(
    threshold_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
)

for cnt in contours:
  mask = np.zeros_like(img)
  cv2.drawContours(mask, [cnt], -1, color=(255, 255, 255), thickness=-1)
  masked = img & mask

  colors = masked.reshape(-1, 3)
  colors = colors[(colors != 0).all(axis=1)]

  k = 1
  criteria = cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0
  ret, label, center = cv2.kmeans(
      colors.astype(np.float32), k, None, criteria, 10, cv2.KMEANS_RANDOM_CENTERS
  )

  # 画像を表示する。
  ax1.imshow(cv2.cvtColor(masked, cv2.COLOR_BGR2RGB))
  ax1.set_axis_off()

  print("center", center)
使用画像

イメージ説明

発生するエラー

center [[113.]]
---------------------------------------------------------------------------
error                                     Traceback (most recent call last)
<ipython-input-19-37cc64ede3fd> in <module>()
     34   criteria = cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0
     35   ret, label, center = cv2.kmeans(
---> 36       colors.astype(np.float32), k, None, criteria, 10, cv2.KMEANS_RANDOM_CENTERS
     37   )
     38 

error: OpenCV(4.1.2) /io/opencv/modules/core/src/kmeans.cpp:241: error: (-2:Unspecified error) in function 'double cv::kmeans(cv::InputArray, int, cv::InputOutputArray, cv::TermCriteria, int, int, cv::OutputArray)'
> Number of clusters should be more than number of elements (expected: 'N >= K'), where
>     'N' is 0
> must be greater than or equal to
>     'K' is 1

補足

Colaboratoryで特になにもカスタムなどせず上記を動かしています。
エラー以外でもここの処理はこうした方がいい、みたいのがあったら指摘してくださると嬉しいです。

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

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

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

    クリップを取り消します

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

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

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

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

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

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

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

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

    質問の評価を下げる

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

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

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

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

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

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

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

    詳細な説明はこちら

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

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

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

まだ回答がついていません

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

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

関連した質問

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