処理した画像の出力、保存を行いたい

前提

最後の画像出力、保存のところがうまくいかない

実現したいこと

rgb画像を各成分毎にヒストグラムの平坦化を行いたい

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

name 'eqh_rgb' is not defined

該当のソースコード

import cv2 import sys import numpy as np img = cv2.imread('中間の画像.jpg') # ヒストグラム平坦化でコントラスト強調 def eqh(img): # 各色で平坦化 b1,g1,r1 = cv2.split(img) b2 = cv2.equalizeHist(b1) g2 = cv2.equalizeHist(g1) r2 = cv2.equalizeHist(r1) eqh_rgb = cv2.merge((b2,g2,r2)) return eqh_rgb cv2.imwrite("output",eqh_rgb)

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）