Pythonでスケルトン処理（細線化）ができない

Question

取り組み内容

Pythonを使って白黒画像を細線化処理したいのですが、以下のコードを実行しても全く動きません。
（すみません、エラーが出たので追記します。）
実行環境はjupyternotebookです。
imgとして読み込んでいる画像は、色抽出を行った後のマスク画像（白黒）です。

https://code.tiblab.net/python/opencv/img_skeletonize
←こちらのサイトを参考にさせていただきました。

import cv2
import numpy as np
from skimage.morphology import skeletonize

img = cv2.imread('desktop/pictures/g3.jpeg')

#細線化(スケルトン化)
ske = skeletonize(~(img != 0))
ske_gray = (ske * 255).astype(np.uint8)
ske_rgb = cv2.cvtColor(ske_gray, cv2.COLOR_GRAY2RGB)

#結果の表示
cv2.imwrite('desktop/pictures/2w.jpg',ske_rgb)

エラー内容

error                                     Traceback (most recent call last)
<ipython-input-14-94446e9f62b4> in <module>
      8 ske = skeletonize(~(img != 0))
      9 ske_gray = (ske * 255).astype(np.uint8)
---> 10 ske_rgb = cv2.cvtColor(ske_gray, cv2.COLOR_GRAY2RGB)
     11 
     12 #結果の表示

error: OpenCV(4.5.2) c:\users\runneradmin\appdata\local\temp\pip-req-build-_8k9tw8n\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<1,-1,-1>,struct cv::impl::A0x57b9ffbe::Set<3,4,-1>,struct cv::impl::A0x57b9ffbe::Set<0,2,5>,2>::CvtHelper(const class cv::_InputArray &,const class cv::_OutputArray &,int)'
> Invalid number of channels in input image:
>     'VScn::contains(scn)'
> where
>     'scn' is 3

Comments

[can110]

提示コードをJupyter上で実行しましたが、現象が再現しません。
フリーズした状態とは、具体的に何がどのようになっているでしょうか？
また、result.jpegはできているでしょうか？
（そもそもですが、画像表示していないので、最後の２行は不要ですが…）

[hakubisin3]

実行した際に左に＊が表示されるだけで、いつまで経っても動作が終了しない状態です。もちろんエラーも出ません。
動かないのでresult.jpegも作成できていません。

[hakubisin3]

先ほど少し変えてから実行すると上記のエラーが発生しました。

Answer 1

スケルトン化はグレースケール画像でないと動かないはずです。scn:3のくだりはそのエラーだと思われます。

なので対策は、
img = cv2.imreadの後に

if len(img.shape())==3:
  img=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

みたいな処理を挟んで3chカラー画像 をグレースケールに変換してください。