【Python】画像の余白を自動的にトリミングするスクリプトを作成する

ポケモンのツールを作るうえで、大量の画像をトリミングしないといけないのですが、

流石に全部を手動トリミングしていると1日あっても終わらないので、そういうスクリプトをPythonで組んでみました。

せっかくなので公開します。

目的
スクリプト作成
まとめ

目的

画像ファイルを読み込んで、余白部分を切り取りたい。

・・・たぶん何を言っているのか分からないと思うので例をあげると、

こういう画像があるとして、

この緑枠線内にトリミングしたいって感じです。

スクリプト作成

必要モジュールのインストール

まずは画像をいじくるためのモジュールであるPILをインストールします。

# pip3 install pillow

python3で使用したかったので、pip3コマンドでインストール。

アルゴリズム検討

簡単に書くと、

上下左右、一番端っこにある色付きピクセルの座標を記録しトリミングする

感じです。

単純ですが、一番わかりやすい方法だと思います。

というか、頭悪いのでこれ以上の方法が思い浮かばんかったです。

スクリプト内容

実際に作成したスクリプトはこちら。

# coding:utf-8

from PIL import Image
import sys
import os

args = sys.argv

outputDir = './trim' #outputディレクトリ

# トリミングする関数
def trimImage(_image, _filename):
  img = _image
  rgbImage = img.convert('RGB')
  size = rgbImage.size
  
  tPix = 100000000
  bPix = -1
  lPix = -1
  rPix = -1
  
  # ピクセル操作
  for x in range(size[0]):
    for y in range(size[1]):
      r,g,b = rgbImage.getpixel((x, y))
      rr,rg,rb = rgbImage.getpixel((size[0] - x-1, size[1] - y-1))
      
      # 色付きのピクセルかどうか（白もしくは白に近しい色を切り抜くため）
      if (r + g + b) < 600:
        if lPix == -1:
          lPix = x
        if y < tPix:
          tPix = y
      
      if (rr + rg + rb) < 600:
        if rPix == -1:
          rPix = size[0] - x
        if size[1] - y > bPix:
          bPix = size[1] - y
  
  print(tPix, lPix, bPix, rPix)
  
  try:
    trimImageFile = img.crop((lPix, tPix, rPix, bPix)) #トリミング
    trimImageFile.save(outputDir + '/' + _filename, quality = 100) #保存
  except:
    print('Error')
          

def main():
  # 出力先ディレクトリがなければ作成する
  if not os.path.exists(outputDir):
    os.mkdir(outputDir)
    
  trimImage(Image.open(args[1]), 'trim_' + args[1])
  
if __name__ == "__main__":
  main()

まずはmain関数から

def main():
  # 出力先ディレクトリがなければ作成する
  if not os.path.exists(outputDir):
    os.mkdir(outputDir)
    
  trimImage(Image.open(args[1]), 'trim_' + args[1])
  
if __name__ == "__main__":
  main()

出力先のディレクトリがなければ作成します。

そのあと、後述するトリミングを実行する関数にImageファイルを食わせます。

ef trimImage(_image, _filename):
  img = _image
  rgbImage = img.convert('RGB')
  size = rgbImage.size

imageオブジェクトと出力する際のファイル名を引数にして実行

ごちゃごちゃ書いているのは画像をRGB変換したり、画像のサイズを取得したりです。

  tPix = 100000000
  bPix = -1
  lPix = -1
  rPix = -1

上下左右の端の座標を記録する用の変数

tPixは大きい数字であれば何でもいいです。

9999999999999999999999999

とかでもよいです。

  for x in range(size[0]):
    for y in range(size[1]):
      r,g,b = rgbImage.getpixel((x, y))
      rr,rg,rb = rgbImage.getpixel((size[0] - x-1, size[1] - y-1))

for文を使用してピクセルを1つずつずらして参照していきます。

r,g,bという変数にその座標のRGB値を代入。

逆側からも同時に参照するためにrr,rg,rbという変数も用意

      # 色付きのピクセルかどうか（白もしくは白に近しい色を切り抜くため）
      if (r + g + b) < 600:
        if lPix == -1:
          lPix = x
        if y < tPix:
          tPix = y
      
      if (rr + rg + rb) < 600:
        if rPix == -1:
          rPix = size[0] - x
        if size[1] - y > bPix:
          bPix = size[1] - y

ここで色付きのピクセルかどうかを判定しています。

完全な白であればr+g+bが765になるのですが、今回は幅を持たせて600以上であれば白と判断しました。

色付きのピクセルであると判断された場合はその座標を記録していきます。

  try:
    trimImageFile = img.crop((lPix, tPix, rPix, bPix)) #トリミング
    trimImageFile.save(outputDir + '/' + _filename, quality = 100) #保存
  except:
    print('Error')

気持ち程度の例外処理をして、画像をトリミング＆保存。