Python: OpenCVを使って画像に集中線を入れる

2018/08/25

OpenCVを使うと画像編集ができます。いろいろな編集ができるみたいですが、今回はよくブログで使われるような集中線が入った画像を作ってみたいと思います。

PythonにOpenCVを導入する方法は以下を見てください。

OpenCVをインストールしてPythonで画像を読みこむテストをしたので備忘録として書きます。今回は単純に画像表示のみ。後々、画像にいろいろなフィルターをかけたり編集していきたいのでその前段階です。まずはOpenCVのインストールOpenCVは以下のコマンドでインストールできます。pip install opencv-python こんな感じでインストールが進みます。サンプルコード画像を表示させるソースコードは以下の通りです。OpenCVをインポートするには import cv2 となります。import cv2pic = cv2.imread("test.jpg")cv2.im...

[ad#top-1]

1 実行例
2 ソースコード
3 解説

実行例

今回紹介するプログラムは、以下のように画像に集中線を入れるものです。

以下がカスタマイズ可能です。

ラインの太さ
ラインの色
画像の中に描く内円の半径
ラインを何°ごと描くか
乱れのレベル

ソースコード

以下がソースコード全文です。5行目から16行目までがパラメータの設定なので必要に応じて修正してください。

import cv2
import math
import random

#画像ファイル名
img_filename = "test.jpg"
#ラインの太さ
border_width = 2
#ラインの色
border_color = (0, 0, 0)
#画像の中に描く内円の半径
r1 =200
#ラインを何°ごと描くか
step = 2
#乱れレベル(0:乱れなし)
sc = 10

#画像を取得
pic = cv2.imread(img_filename)

#画像の高さと幅
h, w = pic.shape[:2]
#中央座標
center_x = int(w * 0.5)
center_y = int(h * 0.5)
#画像の角を接する円の半径
r0 = int(math.sqrt(h ** 2 + w ** 2)) + 1

for theta in range(360):
  if theta % step != 0:
    continue
  #線の乱れ係数
  scattering = random.uniform(0, sc*10)
  #中央を軸にした円の円周座標
  x0 = int(r0 * math.sin(theta * math.pi/180)) + center_x
  y0 = int(r0 * math.cos(theta * math.pi/180)) + center_y
  #描く線の先の座標
  r1a = r1 + scattering
  x1 = int(r1a * math.sin(theta * math.pi/180)) + center_x
  y1 = int(r1a * math.cos(theta * math.pi/180)) + center_y
  #集中線を描く
  cv2.line(pic, (x0, y0), (x1, y1), border_color, border_width)

cv2.imshow('test', pic)

key = cv2.waitKey(0)
if key == 27: #ESCで終了
  cv2.destroyAllWindows()
  exit()
if key == 115: #「s」で保存して終了
  cv2.imwrite('out.jpg',pic)
  cv2.destroyAllWindows()
  exit()