Skip to content

A unified GUI for vision model training, ONNX export, and cross-platform deployment. (開発中)

License

Notifications You must be signed in to change notification settings

kancheng/capybara

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Capybara

Under development | In Entwicklung | En développement | 開発中 | 開發中

Capybara Logo

A unified GUI for vision model training, ONNX export, and cross-platform deployment.

統一的視覺模型訓練、ONNX 匯出和跨平台部署 GUI 工具

License Qt Python Platform


📖 目錄 / Table of Contents

🎯 簡介 / Introduction

Capybara 是一個專為機器學習工作流程設計的綜合性 GUI 工具,從環境配置到模型部署,提供一站式的解決方案。它簡化了視覺模型訓練、數據處理和模型匯出的複雜流程,讓開發者能夠更專注於模型本身。

Capybara is a comprehensive GUI tool designed for machine learning workflows, providing a one-stop solution from environment configuration to model deployment. It simplifies the complex processes of vision model training, data processing, and model export, allowing developers to focus more on the models themselves.

核心價值 / Core Value

  • 🚀 簡化工作流程 / Simplified Workflow:從數據到部署的完整流程
  • 🎨 直觀的用戶界面 / Intuitive UI:易於使用的圖形界面
  • 🔧 自動化環境管理 / Automated Environment Management:智能檢測和管理 Python 環境
  • 📦 多格式支持 / Multi-format Support:支持多種數據格式和模型格式
  • 🌐 跨平台部署 / Cross-platform Deployment:支持多種部署目標

✨ 功能特點 / Features

✅ 已實現 / Implemented

  • 環境設定模塊 / Environment Setup Module
    • ✅ 自動檢測 Python 環境(PATH、Python Launcher、常見位置)
    • ✅ Conda 環境支持(自動檢測和顯示)
    • ✅ 虛擬環境管理
    • ✅ PyTorch 和 Ultralytics 依賴檢測
    • ✅ CUDA 和 GPU 檢測
    • ✅ 設定保存和自動載入

🚧 規劃中 / In Planning

  • 數據處理模塊 / Data Processing Module

    • 數據導入和預處理
    • 數據增強
    • 數據統計分析
  • 數據標註模塊 / Data Annotation Module

    • 圖像標註工具(邊界框、多邊形、關鍵點)
    • 多格式支持(YOLO、COCO、Pascal VOC)
  • 數據格式轉換模塊 / Data Format Conversion Module

    • 標註格式轉換
    • 批量轉換功能
  • 模型訓練模塊 / Model Training Module

    • PyTorch 和 YOLO 訓練支持
    • 訓練監控和可視化
    • 超參數配置
  • 模型匯出模塊 / Model Export Module

    • ONNX 匯出
    • TensorRT、CoreML、TensorFlow Lite 支持
    • 模型優化和量化

💻 系統要求 / System Requirements

最低要求 / Minimum Requirements

  • 操作系統 / Operating System:Windows 10/11 或更高版本
  • Python:3.8 或更高版本
  • 內存 / RAM:至少 4GB
  • 硬盤空間 / Disk Space:至少 500MB

推薦配置 / Recommended Configuration

  • 操作系統 / Operating System:Windows 11
  • Python:3.10 或更高版本
  • 內存 / RAM:8GB 或更多
  • GPU:NVIDIA GPU(支持 CUDA,用於 GPU 加速)
  • 硬盤空間 / Disk Space:至少 2GB(包含模型和數據集)

依賴項 / Dependencies

  • Qt 6.10.1 或更高版本
  • OpenCV(可選,用於圖像處理功能)- 可使用 .\install_opencv.ps1 自動安裝
  • PyTorch(可選,用於模型訓練)
  • Ultralytics(可選,用於 YOLO 模型)
  • CUDA(可選,用於 GPU 加速)

📦 安裝指南 / Installation

從源代碼構建 / Build from Source

  1. 克隆倉庫 / Clone Repository

    git clone https://github.com/kancheng/capybara.git
    cd capybara
  2. 安裝 Qt / Install Qt

    • 下載並安裝 Qt 6.10.1 或更高版本
    • 確保 Qt Creator 或 CMake 已安裝
  3. 安裝 OpenCV(可選,用於圖像處理功能) / Install OpenCV (Optional, for image processing features)

    自動安裝(推薦) / Auto-install (Recommended):

    .\install_opencv.ps1

    或使用 vcpkg(如果已安裝):

    .\install_opencv_simple.ps1

    詳細說明請參閱 INSTALL_OPENCV.md

    For detailed instructions, see INSTALL_OPENCV.md

  4. 配置 CMake / Configure CMake

    mkdir build
    cd build
    cmake ..

    注意 / Note: 如果已安裝 OpenCV,CMake 會自動檢測。如果未安裝,圖像處理功能將被禁用。 If OpenCV is installed, CMake will automatically detect it. If not installed, image processing features will be disabled.

  5. 編譯 / Build

    cmake --build .
  6. 運行 / Run

    ./capybara.exe

使用 Qt Creator

  1. 打開 Qt Creator
  2. 選擇「打開項目」/ "Open Project"
  3. 選擇 CMakeLists.txt
  4. 配置構建設置
  5. 點擊「構建」/ "Build" 和「運行」/ "Run"

🚀 快速開始 / Quick Start

1. 啟動應用程式 / Launch Application

啟動 Capybara 後,應用程式會自動掃描系統中的所有 Python 環境。

After launching Capybara, the application will automatically scan all Python environments on your system.

2. 配置環境 / Configure Environment

  1. 選擇 Python 環境 / Select Python Environment

    • 從「Python 環境」下拉選單中選擇您要使用的 Python 版本
    • Select the Python version you want to use from the "Python Environment" dropdown
  2. 選擇虛擬環境 / Select Virtual Environment

    • 從「虛擬環境」下拉選單中選擇您的 Conda 環境或虛擬環境
    • Select your Conda environment or virtual environment from the "Virtual Environment" dropdown
  3. 指定環境 / Specify Environment

    • 點擊「指定此環境」按鈕保存您的選擇
    • Click the "指定此環境" button to save your selection
    • 應用程式會自動檢測 PyTorch 和 Ultralytics 的安裝狀態
    • The application will automatically detect PyTorch and Ultralytics installation status

3. 查看系統信息 / View System Information

應用程式會自動顯示:

  • 系統架構(x64/x86)
  • CUDA 版本(如果已安裝)
  • GPU 信息(如果可用)

The application automatically displays:

  • System architecture (x64/x86)
  • CUDA version (if installed)
  • GPU information (if available)

📚 使用手冊 / Documentation

詳細的使用手冊和 API 文檔請參閱 document 目錄

For detailed user manual and API documentation, please refer to the document directory.

模塊文檔 / Module Documentation

📁 項目結構 / Project Structure

capybara/
├── document/                 # 文檔目錄 / Documentation directory
│   ├── README.md            # 使用手冊主頁 / User manual homepage
│   ├── environment-setup.md # 環境設定指南 / Environment setup guide
│   ├── environment-setup-details.md # 環境設定詳細說明 / Detailed documentation
│   ├── data-processing.md   # 數據處理文檔 / Data processing docs
│   ├── data-annotation.md   # 數據標註文檔 / Data annotation docs
│   ├── data-conversion.md   # 格式轉換文檔 / Format conversion docs
│   ├── model-training.md    # 模型訓練文檔 / Model training docs
│   └── model-export.md      # 模型匯出文檔 / Model export docs
├── icon/                    # 圖標資源 / Icon resources
│   ├── icon.png            # 項目 LOGO / Project logo
│   ├── raw.png             # 原始圖標 / Raw icon
│   └── raw.ico             # Windows 圖標 / Windows icon
├── build/                   # 構建目錄 / Build directory
├── main.cpp                 # 主程序入口 / Main entry point
├── mainwindow.h             # 主視窗頭文件 / Main window header
├── mainwindow.cpp           # 主視窗實現 / Main window implementation
├── mainwindow.ui            # UI 設計文件 / UI design file
├── resources.qrc            # Qt 資源文件 / Qt resource file
├── capybara.rc              # Windows 資源文件 / Windows resource file
├── CMakeLists.txt           # CMake 構建配置 / CMake build configuration
├── install_opencv.ps1      # OpenCV 自動安裝腳本 / OpenCV auto-install script
├── install_opencv_simple.ps1 # OpenCV vcpkg 安裝腳本 / OpenCV vcpkg install script
├── INSTALL_OPENCV.md        # OpenCV 安裝指南 / OpenCV installation guide
├── LICENSE                  # 許可證文件 / License file
└── README.md               # 本文件 / This file

🛠️ 開發 / Development

技術棧 / Technology Stack

  • 框架 / Framework:Qt 6.10.1
  • 語言 / Language:C++17
  • 構建系統 / Build System:CMake 3.16+
  • 編譯器 / Compiler:MinGW-w64 (GCC)

開發環境設置 / Development Environment Setup

  1. 安裝開發工具 / Install Development Tools

    • Qt Creator 或 Visual Studio
    • CMake
    • Git
  2. 配置 Qt / Configure Qt

    • 確保 Qt 6.10.1 或更高版本已安裝
    • 配置環境變數
  3. 克隆並構建 / Clone and Build

    git clone https://github.com/yourusername/capybara.git
    cd capybara
    mkdir build && cd build
    cmake ..
    cmake --build .

代碼風格 / Code Style

  • 使用 Qt 編碼規範
  • 雙語註釋(中文/英文)
  • 遵循 C++17 標準

構建配置 / Build Configuration

項目使用 CMake 進行構建管理。主要配置選項:

The project uses CMake for build management. Main configuration options:

  • CMAKE_AUTOUIC ON:自動處理 UI 文件
  • CMAKE_AUTOMOC ON:自動處理 Qt MOC
  • CMAKE_AUTORCC ON:自動處理資源文件

🤝 貢獻 / Contributing

我們歡迎所有形式的貢獻!請遵循以下步驟:

We welcome all forms of contributions! Please follow these steps:

  1. Fork 本倉庫
  2. 創建您的特性分支 (git checkout -b feature/AmazingFeature)
  3. 提交您的更改 (git commit -m 'Add some AmazingFeature')
  4. 推送到分支 (git push origin feature/AmazingFeature)
  5. 開啟一個 Pull Request

貢獻指南 / Contribution Guidelines

  • 確保代碼遵循項目的編碼風格
  • 添加適當的註釋和文檔
  • 編寫清晰的提交信息
  • 測試您的更改

📄 許可證 / License

本項目採用 MIT 許可證。詳見 LICENSE 文件。

This project is licensed under the MIT License. See the LICENSE file for details.

🙏 致謝 / Acknowledgments

📞 聯繫 / Contact

如有問題或建議,請通過以下方式聯繫:

For questions or suggestions, please contact:


Made with ❤️ by the Capybara Team

返回頂部 | 文檔 | 許可證

About

A unified GUI for vision model training, ONNX export, and cross-platform deployment. (開発中)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published