zhenghu
|
8a2ea1abd7
|
fix: 使用 ViT 专用类加载模型，移除 Auto 类兼容回退逻辑
- 将 AutoImageProcessor / AutoModelForImageClassification 替换为
ViTImageProcessor / ViTForImageClassification
- 移除 ValueError 异常捕获回退逻辑，改为直接使用 ViT 类加载，
避免因 preprocessor_config.json 中旧版 image_processor_type
导致 Auto 类无法识别的问题
|
2026-04-15 15:09:17 +08:00 |
|
zhenghu
|
581630c4e0
|
fix: 兼容模型加载失败场景，手动构造 image-classification pipeline
- 增加模型加载的异常处理，当 pipeline 直接加载失败时回退为
手动加载 AutoImageProcessor 和 AutoModelForImageClassification
- 添加 torchvision 依赖
|
2026-04-15 14:27:58 +08:00 |
|
zhenghu
|
86541eb55e
|
feat: 初始化香蕉叶病害智能识别系统
基于 ViT 视觉变换器实现香蕉叶片病害分类应用，支持 7 类病害识别。
- app.py: Streamlit 前端应用，包含图片上传、病害识别、置信度图表展示及防治建议
- pyproject.toml: 项目配置，声明 Python 3.14 及依赖（streamlit、transformers、torch、plotly 等）
- uv.lock: 依赖版本锁定文件
- Dockerfile: 基于 uv 镜像的容器化部署配置，含健康检查
- justfile: 常用开发任务脚本（run、install、format、check 等）
- main.py: 命令行入口
- .doc/: 模型使用说明文档
模型来源: Dmitry43243242/banana-disease-leaf-model (HuggingFace)
|
2026-04-15 11:25:57 +08:00 |
|
|
|
f939f415a9
|
Initial commit
|
2026-04-15 02:40:59 +00:00 |
|
