画像処理での_src.type() == CV_8UC1のエラー対処方法について

Question

前提・実現したいこと

画像処理をしています。ガンマ補正の処理後、コントラストの調整をしたいのですが、このエラーが出てしまい困っています。
コントラストの補正をするには前段階の処理であるガンマ補正後、毎回画像を書き出した画像をグレースケールで読み込まなければ処理できないのでしょうか？

書き出したものを毎回読み込むと処理が遅くなるのではないかと思っており、書き出さずに動かせる何かいい方法をご存じであれば教えていただけると助かります。よろしくお願いいたします。

###追記
参考にさせていただいたサイトより、訂正したところチャンネルが異なるとエラーが出てしまいました。解決方法があればお教えいただきたいです。

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

error                                     Traceback (most recent call last)
<ipython-input-12-ae6612370064> in <module>
     23 
     24 #contrast
---> 25 lab = cv2.cvtColor(img2, cv2.COLOR_BGR2LAB)
     26 lab_planes = cv2.split(img2)
     27 clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8,8))

error: OpenCV(4.4.0) c:\users\appveyor\appdata\local\temp\1\pip-req-build-71670poj\opencv\modules\imgproc\src\color.simd_helpers.hpp:92: error: (-2:Unspecified error) in function '__cdecl cv::impl::`anonymous-namespace'::CvtHelper<struct cv::impl::`anonymous namespace'::Set<3,4,-1>,struct cv::impl::A0x2b0558ac::Set<3,-1,-1>,struct cv::impl::A0x2b0558ac::Set<0,5,-1>,2>::CvtHelper(const class cv::_InputArray &,const class cv::_OutputArray &,int)'
> Invalid number of channels in input image:
>     'VScn::contains(scn)'
> where
>     'scn' is 1

該当のソースコード

import cv2
import math
import numpy as np
import openpyxl as excel

#backgroundimagecreation
#img read
oimg = cv2.imread('back.jpg',0)
height = oimg.shape[0]
width = oimg.shape[1]
resize_img = cv2.resize(oimg,(int(width*0.5),int(height*0.5)))

#ganma
img_blur = cv2.GaussianBlur(resize_img,(499,499),0)
height = img_blur.shape[0]
width = img_blur.shape[1]
np.clip(height, 1,254)
np.clip(width,1,254)
b = img_blur/255.0
g = np.log(0.5)/np.log(b)
img2 = 255.0*np.power(resize_img/255.0,g)
cv2.imwrite('back-ga.jpg',img2)

#contrast
lab = cv2.cvtColor(img2, cv2.COLOR_BGR2LAB)
lab_planes = cv2.split(img2)
clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8,8))
lab_planes[0] = clahe.apply(lab_planes[0])
lab = cv2.merge(lab_planes)
cl1 = cv2.cvtColor(lab, cv2.COLOR_LAB2BGR)
cv2.imwrite("back-cl.jpg",cl1)

Comments

[jbpb0]

こんなやり方とか
https://stackoverflow.com/questions/25008458/how-to-apply-clahe-on-rgb-color-images

【追記】上記は、画像ファイルをカラーで読んでいる場合の対処法です
ご質問のコードを読み直したら、グレースケールで読んでるのですね
失礼しました

Answer 1

How to apply CLAHE on RGB color images
の回答に、RGBをLabに変換して、Lに処理してからRGBに戻すコードがありますので、参考にしてみてください

【追記】上記は、画像ファイルをカラーで読んでいる場合の対処法です
ご質問のコードを読み直したら、グレースケールで読んでるのですね
失礼しました
その場合は、

clahe.apply(img2.astype(np.uint8))

のように、CV_8UC1(またはCV_16UC1)に変換すれば大丈夫だと思います

cv2.cvtColor(img2, cv2.COLOR_BGR2LAB)

は不要です

Comments

[gomsis]

ありがとうございます。
参考にさせていただきました。しかし、なかなかうまくいきません。
追記しましたので見ていただきたいです。

[jbpb0]

ご質問のコードについて勘違いしてましたので、回答を訂正しました

[gomsis]

ありがとうございます！うまくいきました！