🌿 Plant Recognition

AI-Powered Plant Identification & Care WeChat Mini-Program

拍照即识 · 智能养护 · 植物百科

---

🌱 What is Plant Recognition?

Plant Recognition 是一个基于深度学习的微信小程序，让你只需拍一张照片，就能快速识别植物种类，并获取专业的养护指南。

从识别到养护，一站式管理你的绿色生活。

✨ Features

🔍 智能识别 拍照 / 相册选图一键识别 返回置信度与植物详情 MobileNetV3 + ONNX 毫秒级推理	🌱 养护管理 创建个人植物档案 记录浇水、施肥等养护日志 智能养护提醒通知
📚 植物百科 浏览完整植物图鉴 查看养护知识详情 搜索与分类浏览	👤 个人中心 微信一键登录 识别历史记录 个人资料管理

🏗 Tech Stack

Frontend	Backend	AI / ML	Infra

📐 Architecture

                    ┌──────────────────────────┐
                    │    📱 WeChat Mini-Program  │
                    │  原生开发 + Vant Weapp     │
                    └────────────┬─────────────┘
                                 │ HTTPS / RESTful API
                    ┌────────────▼─────────────┐
                    │   ⚡ FastAPI Backend       │
                    │  Auth │ Recognize │ Care  │
                    └────────────┬─────────────┘
                                 │
              ┌──────────────────┼──────────────────┐
              │                  │                  │
    ┌─────────▼──────┐ ┌────────▼───────┐ ┌───────▼────────┐
    │  🗄 MySQL 8.0   │ │  🔴 Redis 7.x  │ │  🧠 ONNX       │
    │  植物/用户数据   │ │  缓存/会话管理  │ │  模型推理引擎   │
    └────────────────┘ └────────────────┘ └────────────────┘

🚀 Quick Start

Prerequisites

• Python 3.11+
• Node.js 16+
• Docker & Docker Compose
• 微信开发者工具

1. Backend

cd backend

# 🐳 Docker 一键启动 (推荐)
docker-compose up -d

# 📦 或手动启动
pip install -r requirements.txt
cp .env.example .env
uvicorn app.main:app --reload --host 0.0.0.0 --port 8000

📖 API 文档: http://localhost:8000/docs

2. Mini-Program

1. 微信开发者工具导入 miniprogram/ 目录
2. project.config.json 填入你的 AppID
3. app.js 修改 baseUrl 为后端地址

3. Model Training (Optional)

cd ml
pip install -r requirements.txt

python train.py --data_dir ./data/plants --epochs 50
python evaluate.py --model_path ./output/best_model.pth
python export_onnx.py --model_path ./output/best_model.pth --quantize

💡 开发模式：模型文件不存在时自动返回模拟识别结果，无需训练即可调试前端。

📁 Project Structure

PlantRecognition/
├── 📂 backend/                    # FastAPI 后端
│   ├── app/
│   │   ├── api/                   #   API 路由
│   │   ├── models/                #   ORM 模型
│   │   ├── schemas/               #   数据验证
│   │   ├── services/              #   业务逻辑
│   │   ├── core/                  #   数据库 / Redis / JWT
│   │   └── ml/                    #   ONNX 推理
│   ├── docker-compose.yml
│   └── init.sql
├── 📂 miniprogram/               # 微信小程序
│   ├── pages/                     #   17 个页面
│   ├── components/                #   自定义组件
│   └── utils/                     #   工具函数
└── 📂 ml/                        # 模型训练
    ├── train.py
    ├── evaluate.py
    └── export_onnx.py

🔌 API Endpoints

Method	Endpoint	Description
POST	/api/auth/login	微信登录
POST	/api/recognize	上传图片识别植物
GET	/api/plants	植物列表
GET	/api/plants/{id}	植物详情
GET	/api/my-plants	我的植物
POST	/api/my-plants/{id}/logs	添加养护日志
POST	/api/reminders	创建养护提醒

🤝 Contributing

1. Fork this repository
2. Create your feature branch (git checkout -b feature/amazing-feature)
3. Commit your changes (git commit -m 'feat: add amazing feature')
4. Push to the branch (git push origin feature/amazing-feature)
5. Open a Pull Request

📄 License

This project is licensed under the MIT License.

---

Made with 💚 by Sfrui

If you find this project useful, please consider giving it a ⭐