Python:画像を各チャンネルに分けて保存しなおす

Question

前提・実現したいこと

最近学び始めた初心者です。

（実現したいこと）
・Pythonを使用
・アルファチャンネルを持った画像をRGBAの各チャンネルに分離
・チャンネルごとに異なるサフィックスを付けて別名で保存（保存先は元ファイルのあった場所）
上記処理をファイルのドラッグアンドドロップで行いたいです。

特定の画像を指定する場合のコードは書けたのですが
ドラッグアンドドロップで実行、元ファイルのある場所に保存するという処理をどう書けばいいか分かりません。
cv2ではdds形式だと上手くいきませんでしたのでPILを使用しています。

使用している画像フォーマットはdds形式で、保存し直す時にpngに変換したいです。

発生している問題・エラーメッセージ

エラーメッセージ

該当のソースコード

from PIL import Image
import numpy as np
import os.path
           

# 開く
img = Image.open("./XXXXXX_AAA.dds")

# チャンネル分割
r, g, b, a = img.split()

# バラバラに分割格納されるので正しいチャンネルに再格納
img_rgba = Image.merge('RGBA',(g, r, a, b))

# 再度チャンネル分割
r, g, b, a = img_rgba.split()
# 名前と拡張子を分けて記憶
img, ext = os.path.splitext(img.filename)

# 各チャンネルごとにサフィックスを付けて保存＆_AAAを削除
r.save(img.strip("_AAA") + "_01.png")
g.save(img.strip("_AAA") + "_02.png")
b.save(img.strip("_AAA") + "_03.png")
a.save(img.strip("_AAA") + "_04.png")

試したこと

●参考
https://sd08419ttic.hatenablog.com/entry/2019/03/16/145938
こちらのページのコードを変えて似たような事ができるかな？と思ったのですが上手くいきません…

ご教示宜しくお願い致します。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[tiitoi]

変換する画像の引数の指定はコマンドライン引数経由 (argparse を使う) ではダメなのでしょうか？
ドラッグアンドドロップとなると、GUI ライブラリ等使う必要があり、問題がかなり難しくなりますよ

[sum0x0]

変換したい画像がいくつもあるので、pyinstallerでexe化し
複数選択でドラッグアンドドロップでまとめて処理できるようにしたいと考えているのですが
難しいでしょうか…

[meg_]

対象の画像のパス一覧をcsvにして、コマンドライン引数で指定すれば良いのではないでしょうか？

[sum0x0]

目標としてはexeもしくはバッチファイル化して、他人が使うこともできるものを作りたいと思っています。
そのため対象ファイルをドラッグアンドドロップで変換できるのが一番かなと考えていますが、やはりハードル高いのでしょうか…

Answer 1

pyinstaller は使ったことがないのでわかりませんが、一般的な GUI ライブラリを使えば、ドラッグアンドドロップでパスを取得するのは可能です。

以下、PyQt5 を使ったサンプルを作ってみました。

PyQt5 がない場合は pip でインストールしてください。

pip install PyQt5

test.py として保存

# -*- coding: utf-8 -*-
import sys
from PyQt5.QtWidgets import QWidget, QApplication
from PIL import Image
from pathlib import Path


def process_img(path):
    print(f"processing... '{path}'")

    # 画像を読み込む。
    img = Image.open(path)

    # チャンネルごとに分割する。
    for channel, name in zip(img.split(), img.getbands()):
        save_path = path.parent / f"{path.stem}_{name}.png"
        channel.save(save_path)


class Example(QWidget):
    def __init__(self):
        super().__init__()
        self.setGeometry(300, 300, 300, 300)
        # ドロップ操作を可能にする
        self.setAcceptDrops(True)

    def dragEnterEvent(self, e):
        if e.mimeData().hasUrls():
            # ファイルパスの場合は受け入れる。
            e.accept()
        else:
            e.ignore()

    def dropEvent(self, e):
        # QUrl を pathlib.Path に変換する。
        paths = [Path(url.toLocalFile()) for url in e.mimeData().urls()]

        # 各パスに対して処理を実行する。
        for p in paths:
            process_img(p)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    ex.show()
    app.exec_()

python test.py として実行すると、次のウィンドウが表示されます。

画像をドラッグアンドドロップすると、その場所にチャンネルごとに分離した画像が別名で保存されます。

Comments

[sum0x0]

ご回答頂きありがとうございます。
こういう書き方をするのですね。勉強になります。
やりたいことにかなり近づきました。
あとは元のサフィックスを削除して、なぜかdds形式だとRGBAがバラバラに格納されてしまうのでそのあたりをもう少し探ってみます。

[sum0x0]

バラバラに各チャンネルが格納されてしまう件は読み込んだ後に格納し直す事で解決できました。

[sum0x0]

その後色々試行錯誤していたら望む形のものができました。
ご教示頂きありがとうございました！