ImageClassificationGUI

基于 PySide6 和 ONNX Runtime 的图像分类桌面应用

简体中文 | English

预览

快速开始

1. 克隆

git clone https://github.com/huang2fire/ImageClassificationGUI.git

2. 环境

推荐使用 uv 进行依赖管理。

cd ImageClassificationGUI
uv sync

3. 运行

uv run main.py

使用说明

应用配置文件位于 config/app.toml，可自定义界面风格、字体等参数。

应用支持的类别标签 JSON 文件的内容格式如下：

{
    "0": "类别 0",
    "1": "类别 1",
    "2": "类别 2",
    "3": "类别 3",
    "4": "类别 4"
}

应用支持的图像文件格式可在配置文件中定义：

[load]
image_extensions = ["*.bmp", "*.jpg", "*.jpeg", "*.png", "*.tif", "*.tiff"]

应用仅支持 ONNX 格式的模型文件。

许可证

本项目采用 MIT 许可证，详见 LICENSE 文件。