python 黒から白への変換

Question

画像の黒い背景を白にしたいです。

この画像の外側の黒っぽい部分を白に変えたいです。

files = glob.glob("./1212sisa/*")
for i, f in enumerate(files):
    img = cv2.imread(f)
    hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
    h, s, v = cv2.split(hsv)
    v[20 > v ]=0

現状コードはほとんどかけていませんがどなたかよろしくお願いします。

Comments

[2KOH]

この質問は、画像の黒い背景を白にするコードを書いてほしいという作業依頼ですか？

[reonald]

コードを書いて頂けると大変ありがたく思います。

[2KOH]

https://teratail.com/help/avoid-asking
> 何かを作りたいのでコードを書いてほしい、学校の課題を解いてほしい等の質問は、具体的にプログラミングで困っている質問ではないと考え、推奨していません。

[reonald]

現状つまづいているところとしては、上のコードからのh.s.vの配列の結合ですが、
そもそも調べてみたらhsvで黒→白の変換をしてる人があまりいなかったため、私のコードは使えないと思いこのような形の質問になってしまいました。

Answer 1

このぐらい綺麗な画像であれば、大津の2値化で背景と前景を分離できるので、それで2値化したあとに背景の色を白にすればよいと思います。

import cv2
import numpy as np

def change_background(src_path, dst_path):
    img = cv2.imread(src_path)

    # グレースケールに変換
    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

    # 2値化
    ret, binary = cv2.threshold(gray, 2, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)

    # 背景を白にする。
    img[binary == 0] = 255

    cv2.imwrite(dst_path, img)
    
change_background("sample.jpg", "result.png")

Comments

[reonald]

わかりやすい回答ありがとうございます。