pythonを使用した色空間変換プログラム

Question

前提・実現したいこと

VScode上でPython,OpenCV,NumPyを使って入力画像RGBをHSVに変換し、色相hを操作して色を変え出力画像RGBに戻しフォルダに書き込むプログラムを作成中です。
H(色相)を変えるコードはまだありません(プログラム中のHを変えるだけと思っています)。

下記コードでは、読み込んだ画像と出力した画像が同じであれば成功だと考えています。
現在、エラー止まり、、、。

～行いたいコードの流れ～
画像読込み　→　RGB　→　HSV　→　H(色相)変える　→　RGB　→　画像出力

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

これは使えないよ？、、、みたいなことが書いてあるのがなんとなくわかりました。

IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

該当のソースコード

#-*- coding:utf-8 -*-
import cv2
import numpy as np

def ColorChange(src):
    # 高さ・幅・チャンネル数を取得
    h, w, c = src.shape
    
    # 入力画像と同じサイズで出力画像用の配列を生成(中身は空)
    dst = np.empty((h, w, c))

    for y in range(0, h):
        for x in range(0, w):

            # HSV変換(単体での動作確認OK)
            # R, G, Bの値を取得して0～1の範囲内にする
            #opencv使用
            [b, g, r] = src[y][x]/255.0

            # R, G, Bの値から最大値と最小値を計算
            mx, mn = max(r, g, b), min(r, g, b)

            # 最大値 - 最小値
            diff = mx - mn

            # Hの値を計算(色相変換)
            if mx == mn : h = 0
            elif mx == r : h = 60 * ((g-b)/diff)     
            elif mx == g : h = 60 * ((b-r)/diff) + 120  
            elif mx == b : h = 60 * ((r-g)/diff) + 240
            if h < 0 : h = h + 360
            
            # Sの値を計算
            if mx != 0:s = diff/mx       
            else: s = 0

            # Vの値を計算
            v = mx

            # HSV逆変換(wiki公式参照)
            # 計算上のh'(下記h2)を求める
            h2 = h/60

            # 円柱モデルor円錐モデル判断後、cを求める
            if mx != 0 : c = v * s       
            else: c = s

            # 計算上のxを求める(事前処理1)
            x3 = abs(h2 % 2)
            x2 = x3 - 1
            x = c * x2

            # r,g,b値の計算（事前処理2）
            ver_c = (v-c)*1 + c
            ver_x = (v-c)*1 + x 

            # h2値に合わせた処理(rgb値の確定)
            if 0 <= h2 < 1 :
                r = ver_c
                g = ver_x
                b = 0
            elif 1 <= h2 < 2 :
                r = ver_x
                g = ver_c
                b = 0
            elif 2 <= h2 < 3 :
                r = 0
                g = ver_c
                b = ver_x
            elif 3 <= h2 < 4 :
                r = 0
                g = ver_x
                b = ver_c
            elif 4 <= h2 < 5 :
                r = ver_x
                g = 0
                b = ver_c
            elif 5 <= h2 < 6 :
                r = ver_c
                g = 0
                b = ver_x
            else :
                r = 0
                g = 0
                b = 0
            
　　　　　　　# 計算したrgb値をdstリストの3チャネル(opencvではbgr)に格納
            dst[x, y, [0, 1, 2]] = b, g, r


    return dst


# 入力画像の読み込み
img = cv2.imread('rgb_pix.jpg')

# 方法1(NumPyで実装)
rgb = ColorChange(img)
    
# 結果を出力
cv2.imwrite('rgb_pix2.jpg', rgb)

試したこと

rgb値の代入方法(dst[x, y, [0, 1, 2]] = b, g, r)が間違っているのではないかと思い、opencv配列なのかnumpy配列なのかを調べてみて、flortが違う！みたいなエラーが出たので入力画像はcv2を使ってそのrgb値、hwc値を使って計算し、格納するときはpillowとbase64でcv2と同じ画像を読み込み（格納場所は別にしました）、pillow＆ase64 のrgb値にcv2で読み込み計算したrgb値を格納するなど、、、。いろいろあの手この手で試してみたものの解決には至りませんでした。

補足情報（投稿者の実力）

c,html,css,javascliptを大学で、python,java,rubyを独学でかじった程度の実力です。単純なコードのミスはさんざん調べたので無いと思いますが、どうかよろしくおねがいします。

Answer 1

IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

インデックスエラー: インデックスの指定には整数、slices (:), ellipsis (...)、numpy.newaxis (None)、整数または bool 型の配列のみが使用できます。

なので、インデックスの指定がおかしいということがわかります。

x, y を print すると、x は float であることがわかります。
float 型はインデックスの指定には使用できませんので、エラーになっています。

print(type(x), type(y))  # <class 'numpy.float64'> <class 'int'>
dst[x, y, [0, 1, 2]] = b, g, r

x = c * x2 の部分で x を上書きしてしまっているのが原因だと思うので、変数名が重複しないように変更されてはどうでしょうか

Comments

[ikuta819]

ご回答いただいた通り、ｘが重複していたので変更したところ問題なく出力できました。出力画像は元画像と少し色が変わってしまっているので、式を見直したいと思います。
ご回答、ありがとうございました。