# TianNiu

**Repository Path**: wwmmq/tian-niu

## Basic Information

- **Project Name**: TianNiu
- **Description**: 天牛数据集构建及天牛计数系统开发
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-05-21
- **Last Updated**: 2025-06-09

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# TianNiu

### 🌟 项目介绍
本项目专注于天牛数据集构建与天牛计数系统开发，旨在通过计算机视觉技术实现对天牛目标的高效检测与数量统计。
项目基于 YOLOv5 目标检测框架开发，包含数据集构造、模型训练、系统界面开发全流程解决方案，适用于林业害虫监测、生态研究等场景。

### 📂 项目结构
tian-niu/  
├── 📂 code0529/  
│   ├── 📂 data_augmentation/        ← 数据处理模块  
│   │   ├── __init__.py           ← 数据处理模块初始化文件   
│   │   └── beetle_group_generator.py ← 天牛群数据集生成脚本    
│   ├── 📂 model/                    ← 模型相关代码模块    
│   │   ├── 📂 data/                 ← 数据管理子模块       
│   │   │   ├── group/            ← 天牛群数据集数据目录（图像、标签）     
│   │   │   ├── hyps/             ← 模型超参数配置文件目录（.yaml等）    
│   │   │   ├── images/           ← 天牛原始图像数据目录（训练、验证、测试用）     
│   │   │   ├──  labels/           ← 图像对应标签文件目录（如YOLO格式.txt）   
│   │   │   └── test.py           ← 数据测试脚本，校验数据质量      
│   │   ├── 📂 runs/                 ← 训练与验证结果存储子模块    
│   │   │   ├── train/            ← 模型训练阶段产物目录（日志、中间权重、可视化图表等）    
│   │   │   └── val/exp2/         ← 某次验证实验（exp2标识）结果存储目录  
│   │   ├── weights/              ← 模型权重存储子模块     
│   │   ├── evaluate.py           ← 模型评估脚本，计算输出关键指标  
│   │   └── train.py              ← 模型训练主脚本  
│   └── 📂 ui/                       ← 界面模块  
│       └── main_window.py        ← 主窗口UI实现  
├── 📄config.py                     ← 项目全局配置文件，配置各类参数  
├── 📄main.py                       ← 项目主程序入口，整合各模块流程  
└── 📄requirements.txt              ← 项目依赖清单，记录所需Python库及版本  

### 🚀 核心功能
#### 1. 数据处理（code0529/data_augmentation）
通过 beetle_group_generator.py 批量处理图像，自动生成重叠天牛群图像，关键参数（单图天牛数量、最大重叠率、输出尺寸等）可灵活调整。
#### 2. 模型训练（code0529/model）
基于 YOLOv5 预训练权重，支持 yolov5s/m 版本模型。
自动生成训练配置（beetle_data.yaml），可视化训练曲线（results.png）。
输出关键指标：mAP@0.5（检测精度）、精确率、召回率。
#### 3. 检测与计数（code0529/model + ui）
通过 BeetleDetectorApp 类（main_window.py）构建系统主窗口，集成按钮、图像显示区域、计数标签等组件实现用户交互。
结果可视化：在图像显示区域实时渲染检测结果，包含边界框、置信度及天牛数量，以红 - 黄 - 绿渐变颜色标注边界框（置信度从低到高），直观展示检测置信度。

### 🔧 快速开始
#### 1. 环境准备
克隆仓库：  
git clone https://gitee.com/wwmmq/tian-niu.git   
cd tian-niu   
安装依赖：  
pip install -r requirements.txt  
#### 2. 启动系统
python code0529/main.py