【Python】複数の画像を一気にトリミングするスクリプトを作った話

はじまり#

リサちゃん

はあ～、最近髪伸びてきたなあ、切ろっかな～。

135ml

そういえば、画像を一気に切り抜いてくれるツールを作ったのを思い出したわ！今回はそのツールを紹介するぜ！

リサちゃん

おぉ、切ってくれぇ

ツールの紹介#

Anaconda Promptで動かしていきます。

まず、切り抜きたい、トリミングしたい画像たちが入っているフォルダを指定します。

画像がないフォルダを選ぶとスクリプトは何もなく終了です。

１度選んだ範囲で全てトリミングするか、毎回トリミングする範囲を選んでトリミングするかを選びます。今回は「1」にします。（まあ、「1」が出来ることが今回のスクリプトの真骨頂なので。）

表示されているディレクトリがWindowsなのに「/」区切りになっているのが気になりますが、まあ、一旦進んでみましょう。

トリミングした後のファイルを保存するフォルダの名前を入力します。今回は「トリミング後」にします。

入力すると、実際にトリミングする画面になります。画面サイズは決めておらず、縦長の画像なので、最初はこんな風に歪みます。（デフォルトの画面サイズは設定できるんですかねえ？　まあ、使えればいいので、これで良いですが。）

画面も縦長にすれば、まあ見れる感じになりました。

そして、下の４つのスライダーを動かすと、実際にトリムする範囲を指定できます。

上限が青の線、下限がオレンジの線、左限が緑の線、右限が赤の線です。

範囲を選んだら、ウインドウを×ボタンで閉じます。

すると・・・・、できない！！！

なぜ、できないんだ～。っていう過程はとりあえず置いておきます。実はここが日本語になってるのが原因だったりします。

直して正常に動いたものが以下になります。「このディレクトリを見て確認してね。」的なメッセージが出てきて、トリミングした画像たちも保存されています。

気をつけたところ#

ここでは、ソースを組むに当たって気をつけたところを記載しています。

ディレクトリをWindowsの「\」で区切る。#

以下、ソースの抜粋部分です。

1
def DecideNowDir():
2
    '''
3
    nowDir     : String absolutely directory default folder
4
    nowDirPath : String absolutely path selected folder
5
    '''
6
    nowDir = os.path.abspath(os.path.dirname(__file__))
7
    nowDirPath = filedialog.askdirectory(initialdir=nowDir)
8

9
    # Discrimination whether Windows or Mac.
10
    pf = platform.system()
11
    if pf == 'Windows': # OS is Windows
12
        nowDirPath = nowDirPath.replace('/', '\\')
13
    print("Folder's Absolutely Path: {quotation}{folderpath}{quotation}".format(quotation='"',folderpath=nowDirPath))
14
    return nowDirPath

該当箇所は、filedialog.askdirectory()の部分で、これで選択されたファイルのディレクトリはなぜかWindowsでもMacでも”/”区切りとなってしまうみたいなのです。

それで、後続処理に問題が出るかどうかは調べていませんが、どうにも気持ち悪いのでWindowsで動かした時は文字の置換処理が動くようにしています。

ディレクトリに日本語が入っていると処理を行わずに強制終了させる。#

以下、ソースの抜粋部分です。

1
def TrimImage(trimmed_img_ext):
2
    '''
3
    folderList       : List of file filtered with extension in the selected folder.
4
    selectTimes      : String of times to input.
5
    extracted_dir    : String absolutely path of directory has selected file.
6
    img              : Dictionary of pixels of image. (img [height] [width] [color channel])
7
    trimmed_img      : Dictionary of pixels of image.
8
    trimmed_img_ext : String of extension of trimmed_img.
9
    trimmed_img_name : String of filename of trimmed_img.
10
    '''
11
    fileList = FileListGetter.GetFileList(DirEditor.DecideNowDir(),trimmed_img_ext)
12

13
    # Error Handling
14
    if len(fileList) == 0:
15
        print('\nImageEditor exits because of no target files.')
16
        sys.exit(0)
17

18
    # Error Handling
19
    checkStr = re.compile('[\\a-zA-Z0-9\-\_\.\-\s\:\~\^\=]+')
20
    if checkStr.fullmatch(fileList[0]) == None:
21
        print('\nImageEditor exits because of the directory containing shift-jis character.')
22
        sys.exit(0)
23

24
    selectTimes = input('What times do you choosing? ("1" or "every"): ')
25
    while selectTimes != '1' and selectTimes != 'every':
26
        selectTimes = input('Retry. ("1" or "every"): ')
27
    extracted_dir = DirEditor.MakeDirectory(fileList[0])

2個目の「Error Handling」の部分になります。

僕も、この記事を書くために動かして気づいたのですが、この画像切り抜きに使用しているOpenCVというモジュールは、なんと指定したディレクトリに日本語（ちゃんと言うとShift_JIS）が入っていると、画像を読み取れません・・・。

OpenCVを作った人たちは、Shift_JISを文字だと思っていないようです・・・。日本語を文字だと思っていないようです・・・。

この不具合を回避するために、Pillowやnumpyなどのモジュールを使って処理する方法がありましたが、そこまでして日本語を使いたいわけではないので、いっそのこと、このモジュールを使う時には日本語のディレクトリは使用できないようにエラー処理を行わせるようにしました。

実装方法としては、ソース内で正規表現で指定した文字だけで表記されているディレクトリだけを使用できるようにしています。

1
# Error Handling
2
checkStr = re.compile('[\\a-zA-Z0-9\-\_\.\-\s\:\~\^\=]+')
3
if checkStr.fullmatch(fileList[0]) == None:
4
    print('\nImageEditor exits because of the directory containing shift-jis character.')
5
    sys.exit(0)