🐞 fix & ✨ feat: ヘッダー画像生成処理の改善とエラー処理、ログ出力機能の追加

- アルファチャンネルの処理を改善し、マスク画像の適用方法をより正確にしました。
- 入力画像とマスク画像の存在チェックを追加し、エラー時のログ出力を行うようにしました。
- `loguru`ライブラリを追加し、処理過程のログをファイルに出力するようにしました。
- コマンドライン引数による入力、マスク、出力画像パスの指定を可能にしました。
- 処理開始、完了、エラー発生時のログを記録するよう改善しました。
- マスク画像のアルファチャンネルを考慮した減算処理に変更しました。
- マスク画像と入力画像のサイズが異なる場合の処理を追加する必要があります。(今後の課題)
- ログ出力の設定を追加しました。

pic_to_header/core.py

@@ -1,24 +1,63 @@
 import cv2
 import numpy as np
+import argparse
+import os
+from loguru import logger
 
 def process_header_image(input_image_path, mask_image_path, output_image_path):
+    logger.info(f"処理を開始します: 入力画像={input_image_path}, マスク画像={mask_image_path}")
+    
     # 入力画像を読み込む
     input_image = cv2.imread(input_image_path, cv2.IMREAD_UNCHANGED)
+    logger.info(f"入力画像のサイズ: {input_image.shape}")
     
-    # マスク画像を読み込み、グレースケールに変換
-    mask_image = cv2.imread(mask_image_path, cv2.IMREAD_GRAYSCALE)
+    # マスク画像を読み込む（グレースケールではなく、アルファチャンネル付きで）
+    mask_image = cv2.imread(mask_image_path, cv2.IMREAD_UNCHANGED)
+    logger.info(f"マスク画像のサイズ: {mask_image.shape}")
     
     # マスク画像を入力画像と同じサイズにリサイズ
     mask_image = cv2.resize(mask_image, (input_image.shape[1], input_image.shape[0]))
+    logger.info(f"リサイズ後のマスク画像のサイズ: {mask_image.shape}")
     
     # 入力画像にアルファチャンネルがない場合は追加
     if input_image.shape[2] == 3:
+        logger.info("入力画像にアルファチャンネルを追加します")
         input_image = cv2.cvtColor(input_image, cv2.COLOR_BGR2BGRA)
     
-    # マスクを適用
-    input_image[:, :, 3] = mask_image
+    # マスク画像にアルファチャンネルがない場合は追加
+    if mask_image.shape[2] == 3:
+        logger.info("マスク画像にアルファチャンネルを追加します")
+        mask_image = cv2.cvtColor(mask_image, cv2.COLOR_BGR2BGRA)
+    
+    # マスク画像のアルファチャンネルを取得
+    mask_alpha = mask_image[:, :, 3]
+    
+    # 入力画像のアルファチャンネルを更新
+    # マスクのアルファ値を使って元の画像のアルファ値を減算
+    input_image[:, :, 3] = np.maximum(input_image[:, :, 3] - mask_alpha, 0)
     
     # 結果を保存
     cv2.imwrite(output_image_path, input_image)
+    logger.info(f"処理が完了しました。出力画像: {output_image_path}")
 
     return output_image_path
+
+if __name__ == "__main__":
+    logger.add("process.log", rotation="1 MB")  # ログファイルの設定
+
+    parser = argparse.ArgumentParser(description='画像にマスクを適用してヘッダー画像を生成します。')
+    parser.add_argument('input_image', help='入力画像のパス')
+    parser.add_argument('mask_image', help='マスク画像のパス')
+    parser.add_argument('output_image', help='出力画像のパス')
+    args = parser.parse_args()
+
+    if not os.path.exists(args.input_image):
+        logger.error(f"エラー: 入力画像が見つかりません: {args.input_image}")
+    elif not os.path.exists(args.mask_image):
+        logger.error(f"エラー: マスク画像が見つかりません: {args.mask_image}")
+    else:
+        try:
+            result = process_header_image(args.input_image, args.mask_image, args.output_image)
+            logger.success(f"処理が成功しました。結果: {result}")
+        except Exception as e:
+            logger.exception(f"エラーが発生しました: {str(e)}")