Merge pull request #1 from umyuu/feature/0.0.1

01-installation.bat

@@ -0,0 +1,10 @@
+@echo on
+
+cd %~dp0
+
+python -m venv venv  
+call venv\Scripts\activate
+
+pip install -r requirements.txt  
+
+TIMEOUT /T 10

README.md

@@ -1 +1,26 @@
-# SaliencyMapDemo
+# saliencymap
+## Introduction  
+gradioを使って、opencv-contribパッケージのcv2.saliency.StaticSaliencySpectralResidualを呼び出すサンプルプログラムです。  
+私は画像処理についてはまだ初心者ですが、興味を持っています。  
+以下から導入方法です。  
+
+## 1. 仮想環境(venv)に必要ライブラリのインストール  
+- リポジトリ直下にある以下のバッチを実行する。  
+	~~~
+	01-installation.bat
+	~~~
+	> **NOTE**  
+	仮想環境の名称は「venv」で作成します。  
+
+## 2. 実行  
+- リポジトリ直下にある以下のバッチを実行する。  
+	~~~
+	run.bat
+	~~~
+	> **NOTE**  
+	> デフォルトポートは9999です。  
+	> 他のポート番号例えば8888に変更したい場合は、run.bat内の以下の行を書き換えてください。  
+	> python src\app.py --server_port 8888  
+
+## Source code License.  
+[MIT License](LICENSE)  

requirements.txt

@@ -0,0 +1,75 @@
+aiofiles==23.2.1
+altair==5.3.0
+annotated-types==0.6.0
+anyio==4.3.0
+attrs==23.2.0
+certifi==2024.2.2
+charset-normalizer==3.3.2
+click==8.1.7
+colorama==0.4.6
+contourpy==1.2.1
+cycler==0.12.1
+dnspython==2.6.1
+email_validator==2.1.1
+exceptiongroup==1.2.1
+fastapi==0.111.0
+fastapi-cli==0.0.2
+ffmpy==0.3.2
+filelock==3.14.0
+fonttools==4.51.0
+fsspec==2024.3.1
+gradio==4.28.3
+gradio_client==0.16.0
+h11==0.14.0
+httpcore==1.0.5
+httptools==0.6.1
+httpx==0.27.0
+huggingface-hub==0.23.0
+idna==3.7
+importlib_resources==6.4.0
+Jinja2==3.1.3
+jsonschema==4.22.0
+jsonschema-specifications==2023.12.1
+kiwisolver==1.4.5
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+matplotlib==3.8.4
+mdurl==0.1.2
+numpy==1.26.4
+opencv-python==4.9.0.80
+opencv-contrib-python==4.9.0.80
+orjson==3.10.2
+packaging==24.0
+pandas==2.2.2
+pillow==10.3.0
+pydantic==2.7.1
+pydantic_core==2.18.2
+pydub==0.25.1
+Pygments==2.17.2
+pyparsing==3.1.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.0.1
+python-multipart==0.0.9
+pytz==2024.1
+PyYAML==6.0.1
+referencing==0.35.1
+requests==2.31.0
+rich==13.7.1
+rpds-py==0.18.0
+ruff==0.4.2
+semantic-version==2.10.0
+shellingham==1.5.4
+six==1.16.0
+sniffio==1.3.1
+starlette==0.37.2
+tomlkit==0.12.0
+toolz==0.12.1
+tqdm==4.66.4
+typer==0.12.3
+typing_extensions==4.11.0
+tzdata==2024.1
+ujson==5.9.0
+urllib3==2.2.1
+uvicorn==0.29.0
+watchfiles==0.21.0
+websockets==11.0.3

run.bat

@@ -0,0 +1,7 @@
+@echo on
+
+cd %~dp0
+call venv\Scripts\activate
+python src\app.py --server_port 9999
+
+TIMEOUT /T 10

src/app.py

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+"""
+    SaliencyMapDemo
+"""
+import argparse
+import cv2
+import gradio as gr
+import numpy as np
+import sys
+
+PROGRAM_NAME = 'SaliencyMapDemo'
+__version__ = '0.0.1'
+
+def compute_saliency(image: np.ndarray):
+    # OpenCVのsaliencyを作成
+    saliency = cv2.saliency.StaticSaliencySpectralResidual_create()
+    # 画像の顕著性を計算
+    success, saliencyMap = saliency.computeSaliency(image)
+
+    if success:
+        # 顕著性マップをカラーマップに変換
+        saliencyMap = (saliencyMap * 255).astype("uint8")
+        saliencyMap = cv2.applyColorMap(saliencyMap, cv2.COLORMAP_JET)
+        # 元の画像とカラーマップを重ね合わせ
+        overlay = cv2.addWeighted(image, 0.5, saliencyMap, 0.5, 0)
+        return overlay
+    else:
+        return image  # エラーが発生した場合は元の画像を返す
+
+def main(args):    
+    """
+        Entry Point
+    """
+    with gr.Blocks() as demo:
+        gr.Markdown(
+        """
+        # Saliency Map demo.
+        1. inputタブで画像を選択します。
+        2. Submitボタンを押します。※外部送信していません。ローカルで完結しています。
+        3. 結果がoverlayタブに表示されます。
+        """)
+
+        submit_button = gr.Button("submit")
+        
+        with gr.Row():
+            with gr.Tab("input"):
+                image_input = gr.Image()
+            with gr.Tab("overlay"):
+                image_overlay = gr.Image()
+
+        
+        submit_button.click(compute_saliency, inputs=image_input, outputs=image_overlay)
+        sys.version
+        gr.Markdown(
+        f"""
+        Python {sys.version}  
+        App {__version__}  
+        """)
+        demo.queue(default_concurrency_limit=5).launch(max_file_size=args.max_file_size, server_port=args.server_port)
+
+if __name__ == "__main__":
+    """
+        コマンドライン引数の解析
+    """
+    parser = argparse.ArgumentParser(prog=PROGRAM_NAME, description="SaliencyMapDemo")
+    parser.add_argument('--server_port', type=int, default=9999, help="Gradio server port")
+    parser.add_argument('--max_file_size', type=int, default=20 * gr.FileSize.MB, help="Gradio max file size")
+    parser.add_argument('--version', action='version', version='%(prog)s {0}'.format(__version__))
+
+    main(parser.parse_args())
+