OpenCV、python、画像の色抽出

Question

前提・実現したいこと

OpenCVを使った画像の色抽出

発生している問題・エラーメッセージ

imwrite_('/Users/****/apple.pgm'): can't write data: OpenCV(4.1.0) /Users/travis/build/skvark/opencv-python/opencv/modules/imgcodecs/src/grfmt_pxm.cpp:427: error: (-5:Bad argument) Portable bitmap(.pgm) expects gray image in function 'write'

### 該当のソースコード

#!/usr/bin/python
# -*- coding: utf-8 -*-
# 各種ライブラリのインポート
import numpy as np
import cv2

# 画像読み込み
image_file = '/Users/****/apple.ppm'
img = cv2.imread(image_file)

# 赤以外を消す
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
lower = np.array([0, 127, 0])    # 赤色の範囲指定(min)
upper = np.array([30, 255, 255])   # 赤色の範囲指定(max)
img_mask = cv2.inRange(hsv, lower, upper)
img_color = cv2.bitwise_and(img, img, mask=img_mask)

# 出力
cv2.imwrite("/Users/****/apple.pgm", img_color)
```python



### 使用環境

Mac, python2.7, Pycharm

Answer 1

Portable bitmap(.pgm) expects gray image in function 'write'

と言われているので、以下の1行を追加してグレースケールに変換してから保存すればよいです。

dst = cv2.cvtColor(img_color, cv2.COLOR_BGR2GRAY)

import numpy as np
import cv2

# 画像読み込み
img = cv2.imread('apple.jpg')

# 赤以外を消す
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
lower = np.array([0, 127, 0])    # 赤色の範囲指定(min)
upper = np.array([30, 255, 255])   # 赤色の範囲指定(max)
img_mask = cv2.inRange(hsv, lower, upper)
img_color = cv2.bitwise_and(img, img, mask=img_mask)

dst = cv2.cvtColor(img_color, cv2.COLOR_BGR2GRAY)

# 出力
cv2.imwrite("dst.pgm", dst)