Python,OpenCVで二値画像から白と黒の面積比を算出

はじめに

PythonとOpenCVを使って二値画像（白黒画像）の白と黒の面積比を算出する方法を実装。
ソースコードと計算結果を解説します。

二値化処理は画像処理で最も基本的な処理の一つで、色々なことに応用出来ます。
物体検出などは典型的な応用例ですので、興味ある方は下の記事も参考にしてみてください。

YAJIROBE BLOG

2019.05.02

Python,OpenCVを使って物体検出でコイン(小銭）を検出してみた

https://techtech-sorae.com/pythonopencvを使って物体検出でコイン小銭）を検出して

はじめに画像に写っているコイン（小銭）だけを検出してみたいと思います。最近は機械学習やディープラーニングが流行っていますが、今回は原始的な手法でやってみたいと思います。画像処理や画像認識を書籍や動画教材から学びたい方はこちらの記事もあわせてどうぞ！機械学習とかディープラーニングの概要や勉強方法について知りたい人は、下記の記事もオススメです。動作環境・PyCharm Community Edition 2018.3.1 x64・OpenCV 3.4.5.20どうやってコインを検出するかお札やコインが写っている画像の中でどうやってコインを見つ...

動作環境

・PyCharm Community Edition 2018.3.1 x64
・OpenCV 3.4.5.20

二値画像と二値化処理

二値画像とは、画像の色を二色（0:黒、1：白）で表した画像のことです。
ただし、一般的には256階調の画像では画素値0を黒、255を白と表します。

白と黒の面積比計算

画像はピクセルの集まりなので、各ピクセルが黒か白かを調べてそれぞれの総和を計算します。

そして最終的な面積は画像全体のサイズで割ることで割合を算出します。

各ピクセルが0:黒ではない数を数えるのに便利な関数がOpenCVには実装されていて、それがcountNonZero()です。
この関数を使えばfor文でいちいち書かなくても良いので非常に便利です。

ソースコード

import cv2
import numpy as np

def calc_black_whiteArea(bw_image):
    image_size = bw_image.size
    whitePixels = cv2.countNonZero(bw_image)
    blackPixels = bw_image.size - whitePixels

    whiteAreaRatio = (whitePixels/image_size)*100#[%]
    blackAreaRatio = (blackPixels/image_size)*100#[%]

    print("White Area [%] : ", whiteAreaRatio)
    print("Black Area [%] : ", blackAreaRatio)


if __name__ == "__main__":
    # read input image
    image = cv2.imread("./lenna.jpg")

    # convert grayscale image
    gray_image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)

    # black white
    ret, bw_image = cv2.threshold(gray_image, 0, 255, cv2.THRESH_OTSU)

    # save image
    cv2.imwrite("black_white.jpg", bw_image)

    # calculation black and white area
    calc_black_whiteArea(bw_image)