hsi-preprocessing-toolkit 1.1.4__tar.gz → 2.0.0__tar.gz

{hsi_preprocessing_toolkit-1.1.4 → hsi_preprocessing_toolkit-2.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hsi-preprocessing-toolkit
-Version: 1.1.4
+Version: 2.0.0
 Summary: HSI Preprocessing Toolkit
 Author: songyz2023
 Author-email: songyz2023 <songyz2023dlut@outlook.com>
@@ -33,12 +33,14 @@ Description-Content-Type: text/markdown
 ![GitHub License](https://img.shields.io/github/license/songyz2019/hsi-preprocessing-toolkit)
 
 
-A Hyperspectral Preprocessing Toolkit that
+HSI Preprocessing Toolkit (HPT, formerly HDR2MAT) is a hyperspectral image preprocessing toolset that:
 1. Read the raw data from the HSI camera, and convert it into `.mat` file
 2. Read the `.mat` file
 3. Preview HSI, and convert it to RGB `.png` file
 4. Crop and rotate the HSI and preview in realtime
 5. Select spectrals of interest visually and save them into a `.mat` file
+6. Mix multiple HSI images with layers. (coming in v2.0.0)
+7. Some other utils
 
 
 ## Usage

{hsi_preprocessing_toolkit-1.1.4 → hsi_preprocessing_toolkit-2.0.0}/README.md

@@ -10,12 +10,14 @@
 ![GitHub License](https://img.shields.io/github/license/songyz2019/hsi-preprocessing-toolkit)
 
 
-A Hyperspectral Preprocessing Toolkit that
+HSI Preprocessing Toolkit (HPT, formerly HDR2MAT) is a hyperspectral image preprocessing toolset that:
 1. Read the raw data from the HSI camera, and convert it into `.mat` file
 2. Read the `.mat` file
 3. Preview HSI, and convert it to RGB `.png` file
 4. Crop and rotate the HSI and preview in realtime
 5. Select spectrals of interest visually and save them into a `.mat` file
+6. Mix multiple HSI images with layers. (coming in v2.0.0)
+7. Some other utils
 
 
 ## Usage

{hsi_preprocessing_toolkit-1.1.4 → hsi_preprocessing_toolkit-2.0.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hsi-preprocessing-toolkit"
-version = "1.1.4"
+version = "2.0.0"
 description = "HSI Preprocessing Toolkit"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -33,7 +33,7 @@ url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
 default = true
 
 [build-system]
-requires = ["uv_build >= 0.8.0, <0.9.0"]
+requires = ["uv_build >= 0.8.0, <0.10.0"]
 build-backend = "uv_build"
 
 [tool.hatch.build.targets.wheel]

hsi_preprocessing_toolkit-2.0.0/src/hsi_preprocessing_toolkit/__main__.py

@@ -0,0 +1,28 @@
+import matplotlib.pyplot as plt
+import gradio as gr
+import gradio.utils
+import logging
+from .page.scanner_calc import scanner_calc_tab
+from .page.about import about_tab
+from .page.hsi_preprocessing import hsi_preprocessing_tab
+from .constant import i18n
+
+logging.basicConfig(level=logging.INFO)
+
+def main():
+    theme = gr.themes.Default(primary_hue='cyan').set(
+        button_primary_background_fill='#39c5bb',
+        button_primary_background_fill_hover="#30A8A0",
+    )
+
+    with gr.Blocks(title="hsi-preprocessing-toolkit", theme=theme) as demo:
+        hsi_preprocessing_tab()
+        scanner_calc_tab()
+        about_tab()
+
+    demo.launch(share=False, inbrowser=True, i18n=i18n, favicon_path="asset/icon.ico")
+
+
+if __name__ == "__main__":
+    plt.rcParams['font.family'] = 'SimHei'
+    main()

hsi_preprocessing_toolkit-2.0.0/src/hsi_preprocessing_toolkit/algorithm.py

@@ -0,0 +1,44 @@
+from scipy.ndimage import rotate
+import numpy as np
+import logging
+from copy import deepcopy
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+logger.info("started")
+
+# Return A HWC Image, the C should work with both RGB and HSI
+def composite_img(imgs :list[np.ndarray], transforms:list[dict]):
+    imgs = deepcopy(imgs)
+    # Transform
+    logger.info(f"{transforms=}")
+    for i, (img,trans) in enumerate(zip(imgs, transforms)):
+        rotate_deg = trans['rotation']
+        if rotate_deg % 360 != 0:
+            img = rotate(img, angle=rotate_deg, axes=(0, 1), reshape=True)
+
+        # Crop
+        crop_top, crop_left, crop_bottom, crop_right = trans['crop']
+        if crop_top > 0 or crop_left > 0 or crop_bottom > 0 or crop_right > 0:
+            crop_bottom = None if crop_bottom == 0 else -crop_bottom
+            crop_right  = None if crop_right == 0  else -crop_right
+            img = img[crop_top:crop_bottom, crop_left:crop_right, :]
+        
+        imgs[i] = img
+        
+    # Caclate canvas size
+    shapes = [x.shape for x in imgs]
+    offsets = [ x['location'] for x in transforms ]
+    canvas_c = imgs[0].shape[-1]
+    sizes = [ ((h+abs(x)),w+abs(y)) for (h,w,_),(x,y) in zip(shapes, offsets)]
+    canvas_h, canvas_w = ( max([x[0] for x in sizes]), max([x[1] for x in sizes]) )
+    canvas_shape = (canvas_h, canvas_w, canvas_c)
+    logger.info(f"{sizes=} {canvas_shape=} {shapes=}")
+    canvas = np.zeros(shape=canvas_shape, dtype=imgs[0].dtype)
+
+    # 合成 Compositing
+    for img,trans in zip(imgs, transforms):
+        x,y = trans['location']
+        h,w,_ = img.shape
+        canvas[x:x+h, y:y+w] = img
+    return canvas

hsi_preprocessing_toolkit-2.0.0/src/hsi_preprocessing_toolkit/constant.py

@@ -0,0 +1,111 @@
+import gradio as gr
+
+CONSTS = dict(
+    homepage_url = "https://github.com/songyz2019/hsi-preprocessing-toolkit",
+)
+
+i18n = gr.I18n(**{
+    'en': {
+        "about.tab_title": "About",
+        "about.title": "HSI Preprocessing Toolkit",
+        "about.description": "A Hyperspectral Image Preprocessing Toolkit from HSI Camera to Machine Learning Dataset",
+        "about.homepage": "主页",
+
+        "hsi_processing.tab_title": "HSI Processing",
+        "hsi_processing.load": "Load",
+        "hsi_processing.upload_instructions": "**Upload one of the following formats:**\n1. One .hdr file + one raw data file without extension\n2. One .mat file",
+        "hsi_processing.input_format": "Input Image Shape Format",
+        "hsi_processing.data_files": "Data Files",
+        "hsi_processing.manual_normalize": "Manual Normalize (affects preview only)",
+        "hsi_processing.normalize_min": "Normalize Min",
+        "hsi_processing.normalize_max": "Normalize Max",
+        "hsi_processing.wavelength_start": "Wavelength Range Start",
+        "hsi_processing.wavelength_end": "Wavelength Range End",
+        "hsi_processing.processing": "Processing",
+        "hsi_processing.crop": "Crop",
+        "hsi_processing.top": "Top",
+        "hsi_processing.bottom": "Bottom",
+        "hsi_processing.left": "Left",
+        "hsi_processing.right": "Right",
+        "hsi_processing.rotate": "Rotate",
+        "hsi_processing.rotate_degree": "Rotate Degree",
+        "hsi_processing.preview": "Preview",
+        "hsi_processing.apply_processing": "Apply Processing Effects",
+        "hsi_processing.mat_data_type": "MAT Data Type",
+        "hsi_processing.mat_format": "MAT Image Shape Format",
+        "hsi_processing.mat_key": "Key of MAT file",
+        "hsi_processing.compress_mat": "Produce Compressed MAT File",
+        "hsi_processing.spectral_selection": "Spectral Selection",
+        "hsi_processing.spectral_selection_help": "Click on the image to select pixels for spectral data extraction. The selected pixels will be plotted in the spectral plot below.",
+        "hsi_processing.spectral_plot": "Spectral Plot",
+        "hsi_processing.style": "Style",
+        "hsi_processing.clear": "Clear",
+        "hsi_processing.download": "Download",
+        "hsi_processing.output_results": "Output Results",
+        "hsi_processing.mat_file": "MAT File",
+        "hsi_processing.info": "Info",
+        "hsi_processing.same_as_input": "Same",
+        "hsi_processing.auto_detect": "Auto Detect",
+
+        "scanner_calc.tab_title": "Scanner Parameters"
+    },
+    'zh-CN':{
+        "about.tab_title": "关于",
+        "about.title": "HPT高光谱处理工具箱",
+        "about.description": "A Hyperspectral Image Preprocessing Toolkit from HSI Camera to Machine Learning Dataset",
+        "about.homepage": "主页",
+        "about.license": """Copyright (C) 2025  songyz2019
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.""",
+
+        "hsi_processing.tab_title": "高光谱图像处理",
+        "hsi_processing.load": "加载",
+        "hsi_processing.upload_instructions": "**应上传以下两种格式中的一种**\n1. 同时上传一个.hdr文件 + 一个无后缀的数据文件\n2. 一个.mat文件",
+        "hsi_processing.input_format": "输入数据形状",
+        "hsi_processing.data_files": "数据文件",
+        "hsi_processing.manual_normalize": "手动归一化(仅影响预览结果)",
+        "hsi_processing.normalize_min": "归一化最小值",
+        "hsi_processing.normalize_max": "归一化最大值",
+        "hsi_processing.wavelength_start": "波长范围起始",
+        "hsi_processing.wavelength_end": "波长范围结束",
+        "hsi_processing.processing": "处理",
+        "hsi_processing.crop": "裁切",
+        "hsi_processing.top": "上",
+        "hsi_processing.bottom": "下",
+        "hsi_processing.left": "左",
+        "hsi_processing.right": "右",
+        "hsi_processing.rotate": "旋转",
+        "hsi_processing.rotate_degree": "旋转角度",
+        "hsi_processing.preview": "预览",
+        "hsi_processing.apply_processing": "应用处理效果",
+        "hsi_processing.mat_data_type": "mat文件数据类型",
+        "hsi_processing.mat_format": "mat文件格式",
+        "hsi_processing.mat_key": "mat文件的key",
+        "hsi_processing.compress_mat": "启用mat文件压缩",
+        "hsi_processing.spectral_selection": "光谱选择",
+        "hsi_processing.spectral_selection_help": "点击预览图像图像中的像素进行光谱数据提取。选中的像素将在下方的光谱图中绘制。",
+        "hsi_processing.spectral_plot": "光谱图",
+        "hsi_processing.style": "样式",
+        "hsi_processing.clear": "清空",
+        "hsi_processing.download": "下载",
+        "hsi_processing.output_results": "输出结果",
+        "hsi_processing.mat_file": "MAT文件",
+        "hsi_processing.info": "信息",
+        "hsi_processing.same_as_input": "与输入相同",
+        "hsi_processing.auto_detect": "自动检测",
+
+        "scanner_calc.tab_title": "推扫仪计算"
+    }
+})
+

hsi_preprocessing_toolkit-2.0.0/src/hsi_preprocessing_toolkit/page/about.py

@@ -0,0 +1,12 @@
+import gradio as gr
+from ..constant import i18n, CONSTS
+
+def about_tab():
+    with gr.Tab(i18n('about.tab_title')):
+        with gr.Column():
+            gr.Markdown('# ' + i18n('about.title'))
+            gr.Markdown(i18n('about.description'))
+            gr.Markdown(f"[{i18n('about.homepage')}]({CONSTS['homepage_url']})")
+            gr.Markdown("```text\n" + i18n('about.license') + "\n```")
+
+