はじめに

画像に枠線を付ける方法を整理します。

やりたいこと

下記のように単純に画像に枠線を付けることです。

用途ですが、画像の中の物体の輪郭を取るとき、白い枠線があると上手く行くケースがあります。

方式

Numpyのスライスを使います。

例えば、(128, 128, 3)サイズの画像があって、一番下の部分に幅10pxの白いラインを書きたいとします。
すると下記のようなコードになります。


img[128 - 10 : 128 , : , : ] = 255

チャンネルごとに値を入れることをお勧めします。
この方式だとカラーの枠線を入れることが可能です。OpenCVで画像を開いた場合、チャンネル順序が(B,G,R)になっていることには注意してください。


img[128 - 10 : 128 , : , : ] = [255, 255, 255]

このような処理を4辺に繰り返し行います。
draw_white_edge()の関数を使います。

引数として、下記の二つを指定します。

band_width : 枠線の幅、単位はピクセル
color : [B,G,R] 範囲 0~255 整数　

全体コード

簡単ですね。コードを掲示します。

import cv2


def draw_white_edge(img, band_width=10, color = [255, 255, 255]):
    # Image is aligned in BGR order.
    # color or Gray Scale
    if len(img.shape) < 3:  # gray scale
        img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
    else:
        pass

    # Check the image size
    height, width, channel = img.shape[0], img.shape[1], img.shape[2]

    # Image copy
    img_test = img.copy()

    # Image process
    # 1. bottom line
    img_test[height - band_width:height, :, :] = color
    # 2. top line
    img_test[0: band_width, :, :] = color
    # 3. left line
    img_test[:, 0:band_width, :] = color
    # 4. right line
    img_test[:, width - band_width:width, :] = color

    return img_test


def main():
    print('Welcome to my program!')
    img_array = cv2.imread('./images/apple.jpg')

    img_result = draw_white_edge(img=img_array, band_width= 50, color= [255, 0, 255])
    cv2.imshow('original', img_array)
    cv2.imshow('result', img_result)
    cv2.waitKey(0)


if __name__ == '__main__':
    main()

#結果