基于YOLOv12的猫狗品种识别检测系统(YOLOv12深度学习+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型)

 一、项目介绍

随着人工智能与计算机视觉技术的飞速发展,基于深度学习的目标检测技术在图像识别领域展现出巨大的应用潜力。本项目旨在开发一个高精度、高效率的猫狗品种识别与检测系统。系统采用前沿的YOLOv12目标检测算法作为核心架构,该模型在速度和精度上相较于前代版本有显著提升,能够实现对图像或实时视频流中猫狗目标的快速、精准定位与品种分类。

项目构建了一个包含37个猫狗品种的大规模定制数据集(nc=37),共计13,983张标注图像。系统后端基于PyTorch深度学习框架进行模型训练与优化,前端则开发了用户友好的图形化界面(UI),集成了登录注册、图像上传、实时检测、结果可视化。实验结果表明,最终训练所得的模型在独立测试集上达到了优异的平均精度(mAP)95.1%,验证了该系统在实际应用中的有效性与可靠性。本项目为宠物识别、智能安防、动物学研究等领域提供了一个功能完备的技术解决方案和工程实践参考。

目录

 一、项目介绍

二、项目功能展示

2.1 用户登录系统

2.2 检测功能

2.3 检测结果显示

2.4 参数配置

2.5 其他功能

3. 技术特点

4. 系统流程

三、数据集介绍

2. 类别详情 (Class Details)

数据集配置文件

四、项目环境配置

创建虚拟环境

安装所需要库

五、模型训练

训练代码

训练结果

六、核心代码

🔐登录注册验证

🎯 多重检测模式

🖼️ 沉浸式可视化

⚙️ 参数配置系统

✨ UI美学设计

🔄 智能工作流

七、项目源码(视频简介)


基于深度学习YOLOv12的猫狗品种识别检测系统(YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型)_哔哩哔哩_bilibili

基于深度学习YOLOv12的猫狗品种识别检测系统(YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型)

二、项目功能展示

✅ 用户登录注册:支持密码检测和安全性验证。

✅ 三种检测模式:基于YOLOv12模型,支持图片、视频和实时摄像头三种检测,精准识别目标。

✅ 双画面对比:同屏显示原始画面与检测结果。

✅ 数据可视化:实时表格展示检测目标的类别、置信度及坐标。

✅智能参数调节:提供置信度滑块,动态优化检测精度,适应不同场景需求。

✅科幻风交互界面:深色主题搭配动态光效,减少视觉疲劳,提升操作体验。

✅多线程高性能架构:独立检测线程保障流畅运行,实时状态提示,响应迅速无卡顿。

2.1 用户登录系统

  • 提供用户登录和注册功能

  • 用户名和密码验证

  • 账户信息本地存储(accounts.json)

  • 密码长度至少6位的安全要求

2.2 检测功能

  • 图片检测:支持JPG/JPEG/PNG/BMP格式图片的火焰烟雾检测

  • 视频检测:支持MP4/AVI/MOV格式视频的逐帧检测

  • 摄像头检测:实时摄像头流检测(默认摄像头0)

  • 检测结果保存到"results"目录

2.3 检测结果显示

  • 显示原始图像和检测结果图像

  • 检测结果表格展示,包含:

    • 检测到的类别

    • 置信度分数

    • 物体位置坐标(x,y)、

2.4 参数配置

  • 模型选择

  • 置信度阈值调节(0-1.0)

  • IoU(交并比)阈值调节(0-1.0)

  • 实时同步滑块和数值输入框

2.5 其他功能

  • 检测结果保存功能

  • 视频检测时自动保存结果视频

  • 状态栏显示系统状态和最后更新时间

  • 无边框窗口设计,可拖动和调整大小

3. 技术特点

  • 采用多线程处理检测任务,避免界面卡顿

  • 精美的UI设计,具有科技感的视觉效果:

    • 发光边框和按钮

    • 悬停和按下状态效果

    • 自定义滑块、表格和下拉框样式

  • 检测结果保存机制

  • 响应式布局,适应不同窗口大小

4. 系统流程

  1. 用户登录/注册

  2. 选择检测模式(图片/视频/摄像头)

  3. 调整检测参数(可选)

  4. 开始检测并查看结果

  5. 可选择保存检测结果

  6. 停止检测或切换其他模式

三、数据集介绍

  • 类别数量 (nc): 37类。其中包括12个猫的品种和25个狗的品种。

  • 数据总量: 13,983张图像

  • 数据划分

    • 训练集 (Training Set): 12,879 张图像

    • 验证集 (Validation Set): 736 张图像

    • 测试集 (Test Set): 368 张图像

2. 类别详情 (Class Details)

数据集的37个类别名称 (names) 如下所示,涵盖了常见及部分稀有的猫狗品种:

  • 猫 (Cats - 12类)cat-Abyssinian (阿比西尼亚猫), cat-Bengal (孟加拉豹猫), cat-Birman (伯曼猫), cat-Bombay (孟买猫), cat-British_Shorthair (英国短毛猫), cat-Egyptian_Mau (埃及猫), cat-Maine_Coon (缅因猫), cat-Persian (波斯猫), cat-Ragdoll (布偶猫), cat-Russian_Blue (俄罗斯蓝猫), cat-Siamese (暹罗猫), cat-Sphynx (斯芬克斯猫).

  • 狗 (Dogs - 25类)dog-american_bulldog (美国斗牛犬), dog-american_pit_bull_terrier (美国比特斗牛梗), dog-basset_hound (巴吉度猎犬), dog-beagle (比格犬), dog-boxer (拳师犬), dog-chihuahua (吉娃娃), dog-english_cocker_spaniel (英国可卡犬), dog-english_setter (英国塞特犬), dog-german_shorthaired (德国短毛指示犬), dog-great_pyrenees (大白熊犬), dog-havanese (哈瓦那犬), dog-japanese_chin (日本犬), dog-keeshond (荷兰毛狮犬), dog-leonberger (莱昂伯格犬), dog-miniature_pinscher (迷你宾沙犬), dog-newfoundland (纽芬兰犬), dog-pomeranian (博美犬), dog-pug (八哥犬), dog-saint_bernard (圣伯纳犬), dog-samoyed (萨摩耶犬), dog-scottish_terrier (苏格兰梗), dog-shiba_inu (柴犬), dog-staffordshire_bull_terrier (斯塔福郡斗牛梗), dog-wheaten_terrier (软毛麦色梗), dog-yorkshire_terrier (约克夏梗).

数据集配置文件

数据集采用标准化YOLO格式组织:

train: F:\猫狗品种识别检测数据集\train\images
val: F:\猫狗品种识别检测数据集\valid\images
test: F:\猫狗品种识别检测数据集\test\images

nc: 37
names: ['cat-Abyssinian', 'cat-Bengal', 'cat-Birman', 'cat-Bombay', 'cat-British_Shorthair', 'cat-Egyptian_Mau', 'cat-Maine_Coon', 'cat-Persian', 'cat-Ragdoll', 'cat-Russian_Blue', 'cat-Siamese', 'cat-Sphynx', 'dog-american_bulldog', 'dog-american_pit_bull_terrier', 'dog-basset_hound', 'dog-beagle', 'dog-boxer', 'dog-chihuahua', 'dog-english_cocker_spaniel', 'dog-english_setter', 'dog-german_shorthaired', 'dog-great_pyrenees', 'dog-havanese', 'dog-japanese_chin', 'dog-keeshond', 'dog-leonberger', 'dog-miniature_pinscher', 'dog-newfoundland', 'dog-pomeranian', 'dog-pug', 'dog-saint_bernard', 'dog-samoyed', 'dog-scottish_terrier', 'dog-shiba_inu', 'dog-staffordshire_bull_terrier', 'dog-wheaten_terrier', 'dog-yorkshire_terrier']

四、项目环境配置

创建虚拟环境

首先新建一个Anaconda环境,每个项目用不同的环境,这样项目中所用的依赖包互不干扰。

终端输入

conda create -n yolov12 python==3.9

激活虚拟环境

conda activate yolov12

安装cpu版本pytorch

pip install torch torchvision torchaudio

安装所需要库

pip install -r requirements.txt

pycharm中配置anaconda

五、模型训练

训练代码

from ultralytics import YOLO

model_path = 'yolo12s.pt'
data_path = 'data.yaml'

if __name__ == '__main__':
    model = YOLO(model_path)
    results = model.train(data=data_path,
                          epochs=100,
                          batch=8,
                          device='0',
                          workers=0,
                          project='runs',
                          name='exp',
                          )
根据实际情况更换模型
# yolov12n.yaml (nano):轻量化模型,适合嵌入式设备,速度快但精度略低。
# yolov12s.yaml (small):小模型,适合实时任务。
# yolov12m.yaml (medium):中等大小模型,兼顾速度和精度。
# yolov12b.yaml (base):基本版模型,适合大部分应用场景。
# yolov12l.yaml (large):大型模型,适合对精度要求高的任务。
  • --batch 8:每批次8张图像。
  • --epochs 100:训练100轮。
  • --datasets/data.yaml:数据集配置文件。
  • --weights yolov12s.pt:初始化模型权重,yolov12s.pt 是预训练的轻量级YOLO模型。

训练结果

六、核心代码

import sys

import cv2
import numpy as np
from PyQt5.QtWidgets import QApplication, QMessageBox, QFileDialog
from PyQt5.QtCore import QThread, pyqtSignal
from ultralytics import YOLO
from UiMain import UiMainWindow
import time
import os
from PyQt5.QtWidgets import QDialog
from LoginWindow import LoginWindow

class DetectionThread(QThread):
    frame_received = pyqtSignal(np.ndarray, np.ndarray, list)  # 原始帧, 检测帧, 检测结果
    finished_signal = pyqtSignal()  # 线程完成信号

    def __init__(self, model, source, conf, iou, parent=None):
        super().__init__(parent)
        self.model = model
        self.source = source
        self.conf = conf
        self.iou = iou
        self.running = True

    def run(self):
        try:
            if isinstance(self.source, int) or self.source.endswith(('.mp4', '.avi', '.mov')):  # 视频或摄像头
                cap = cv2.VideoCapture(self.source)
                while self.running and cap.isOpened():
                    ret, frame = cap.read()
                    if not ret:
                        break

                    # 保存原始帧
                    original_frame = frame.copy()

                    # 检测
                    results = self.model(frame, conf=self.conf, iou=self.iou)
                    annotated_frame = results[0].plot()

                    # 提取检测结果
                    detections = []
                    for result in results:
                        for box in result.boxes:
                            class_id = int(box.cls)
                            class_name = self.model.names[class_id]
                            confidence = float(box.conf)
                            x, y, w, h = box.xywh[0].tolist()
                            detections.append((class_name, confidence, x, y))

                    # 发送信号
                    self.frame_received.emit(
                        cv2.cvtColor(original_frame, cv2.COLOR_BGR2RGB),
                        cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB),
                        detections
                    )

                    # 控制帧率
                    time.sleep(0.03)  # 约30fps

                cap.release()
            else:  # 图片
                frame = cv2.imread(self.source)
                if frame is not None:
                    original_frame = frame.copy()
                    results = self.model(frame, conf=self.conf, iou=self.iou)
                    annotated_frame = results[0].plot()

                    # 提取检测结果
                    detections = []
                    for result in results:
                        for box in result.boxes:
                            class_id = int(box.cls)
                            class_name = self.model.names[class_id]
                            confidence = float(box.conf)
                            x, y, w, h = box.xywh[0].tolist()
                            detections.append((class_name, confidence, x, y))

                    self.frame_received.emit(
                        cv2.cvtColor(original_frame, cv2.COLOR_BGR2RGB),
                        cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB),
                        detections
                    )

        except Exception as e:
            print(f"Detection error: {e}")
        finally:
            self.finished_signal.emit()

    def stop(self):
        self.running = False


class MainWindow(UiMainWindow):
    def __init__(self):
        super().__init__()

        # 初始化模型
        self.model = None
        self.detection_thread = None
        self.current_image = None
        self.current_result = None
        self.video_writer = None
        self.is_camera_running = False
        self.is_video_running = False
        self.last_detection_result = None  # 新增:保存最后一次检测结果

        # 连接按钮信号
        self.image_btn.clicked.connect(self.detect_image)
        self.video_btn.clicked.connect(self.detect_video)
        self.camera_btn.clicked.connect(self.detect_camera)
        self.stop_btn.clicked.connect(self.stop_detection)
        self.save_btn.clicked.connect(self.save_result)

        # 初始化模型
        self.load_model()

    def load_model(self):
        try:
            model_name = self.model_combo.currentText()
            self.model = YOLO(f"{model_name}.pt")  # 自动下载或加载本地模型
            self.update_status(f"模型 {model_name} 加载成功")
        except Exception as e:
            QMessageBox.critical(self, "错误", f"模型加载失败: {str(e)}")
            self.update_status("模型加载失败")

    def detect_image(self):
        if self.detection_thread and self.detection_thread.isRunning():
            QMessageBox.warning(self, "警告", "请先停止当前检测任务")
            return

        file_path, _ = QFileDialog.getOpenFileName(
            self, "选择图片", "", "图片文件 (*.jpg *.jpeg *.png *.bmp)")

        if file_path:
            self.clear_results()
            self.current_image = cv2.imread(file_path)
            self.current_image = cv2.cvtColor(self.current_image, cv2.COLOR_BGR2RGB)
            self.display_image(self.original_image_label, self.current_image)

            # 创建检测线程
            conf = self.confidence_spinbox.value()
            iou = self.iou_spinbox.value()
            self.detection_thread = DetectionThread(self.model, file_path, conf, iou)
            self.detection_thread.frame_received.connect(self.on_frame_received)
            self.detection_thread.finished_signal.connect(self.on_detection_finished)
            self.detection_thread.start()

            self.update_status(f"正在检测图片: {os.path.basename(file_path)}")

    def detect_video(self):
        if self.detection_thread and self.detection_thread.isRunning():
            QMessageBox.warning(self, "警告", "请先停止当前检测任务")
            return

        file_path, _ = QFileDialog.getOpenFileName(
            self, "选择视频", "", "视频文件 (*.mp4 *.avi *.mov)")

        if file_path:
            self.clear_results()
            self.is_video_running = True

            # 初始化视频写入器
            cap = cv2.VideoCapture(file_path)
            frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
            frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
            fps = cap.get(cv2.CAP_PROP_FPS)
            cap.release()

            # 创建保存路径
            save_dir = "results"
            os.makedirs(save_dir, exist_ok=True)
            timestamp = time.strftime("%Y%m%d_%H%M%S")
            save_path = os.path.join(save_dir, f"result_{timestamp}.mp4")

            fourcc = cv2.VideoWriter_fourcc(*'mp4v')
            self.video_writer = cv2.VideoWriter(save_path, fourcc, fps, (frame_width, frame_height))

            # 创建检测线程
            conf = self.confidence_spinbox.value()
            iou = self.iou_spinbox.value()
            self.detection_thread = DetectionThread(self.model, file_path, conf, iou)
            self.detection_thread.frame_received.connect(self.on_frame_received)
            self.detection_thread.finished_signal.connect(self.on_detection_finished)
            self.detection_thread.start()

            self.update_status(f"正在检测视频: {os.path.basename(file_path)}")

    def detect_camera(self):
        if self.detection_thread and self.detection_thread.isRunning():
            QMessageBox.warning(self, "警告", "请先停止当前检测任务")
            return

        self.clear_results()
        self.is_camera_running = True

        # 创建检测线程 (默认使用摄像头0)
        conf = self.confidence_spinbox.value()
        iou = self.iou_spinbox.value()
        self.detection_thread = DetectionThread(self.model, 0, conf, iou)
        self.detection_thread.frame_received.connect(self.on_frame_received)
        self.detection_thread.finished_signal.connect(self.on_detection_finished)
        self.detection_thread.start()

        self.update_status("正在从摄像头检测...")

🔐登录注册验证

对应文件:LoginWindow.py

# 账户验证核心逻辑
def handle_login(self):
    username = self.username_input.text().strip()
    password = self.password_input.text().strip()
    
    if not username or not password:
        QMessageBox.warning(self, "警告", "用户名和密码不能为空!")
        return
    
    if username in self.accounts and self.accounts[username] == password:
        self.accept()  # 验证通过
    else:
        QMessageBox.warning(self, "错误", "用户名或密码错误!")

# 密码强度检查(注册时)
def handle_register(self):
    if len(password) < 6:  # 密码长度≥6位
        QMessageBox.warning(self, "警告", "密码长度至少为6位!")

🎯 多重检测模式

对应文件:main.py

图片检测

def detect_image(self):
    file_path, _ = QFileDialog.getOpenFileName(
        self, "选择图片", "", "图片文件 (*.jpg *.jpeg *.png *.bmp)")
    if file_path:
        self.detection_thread = DetectionThread(self.model, file_path, conf, iou)
        self.detection_thread.start()  # 启动检测线程

视频检测

def detect_video(self):
    file_path, _ = QFileDialog.getOpenFileName(
        self, "选择视频", "", "视频文件 (*.mp4 *.avi *.mov)")
    if file_path:
        self.video_writer = cv2.VideoWriter()  # 初始化视频写入器
        self.detection_thread = DetectionThread(self.model, file_path, conf, iou)

实时摄像头

def detect_camera(self):
    self.detection_thread = DetectionThread(self.model, 0, conf, iou)  # 摄像头设备号0
    self.detection_thread.start()

🖼️ 沉浸式可视化

对应文件:UiMain.py

双画面显示

def display_image(self, label, image):
    q_img = QImage(image.data, w, h, bytes_per_line, QImage.Format_RGB888)
    pixmap = QPixmap.fromImage(q_img)
    label.setPixmap(pixmap.scaled(label.size(), Qt.KeepAspectRatio))  # 自适应缩放

结果表格

def add_detection_result(self, class_name, confidence, x, y):
    self.results_table.insertRow(row)
    items = [
        QTableWidgetItem(class_name),  # 类别列
        QTableWidgetItem(f"{confidence:.2f}"),  # 置信度
        QTableWidgetItem(f"{x:.1f}"),  # X坐标
        QTableWidgetItem(f"{y:.1f}")   # Y坐标
    ]

⚙️ 参数配置系统

对应文件:UiMain.py

双阈值联动控制

# 置信度阈值同步
def update_confidence(self, value):
    confidence = value / 100.0
    self.confidence_spinbox.setValue(confidence)  # 滑块→数值框
    self.confidence_label.setText(f"置信度阈值: {confidence:.2f}")

# IoU阈值同步  
def update_iou(self, value):
    iou = value / 100.0
    self.iou_spinbox.setValue(iou)

✨ UI美学设计

对应文件:UiMain.py

科幻风格按钮

def create_button(self, text, color):
    return f"""
    QPushButton {{
        border: 1px solid {color};
        color: {color};
        border-radius: 6px;
    }}
    QPushButton:hover {{
        background-color: {self.lighten_color(color, 10)};
        box-shadow: 0 0 10px {color};  # 悬停发光效果
    }}
    """

动态状态栏

def update_status(self, message):
    self.status_bar.showMessage(
        f"状态: {message} | 最后更新: {time.strftime('%H:%M:%S')}"  # 实时时间戳
    )

🔄 智能工作流

对应文件:main.py

线程管理

class DetectionThread(QThread):
    frame_received = pyqtSignal(np.ndarray, np.ndarray, list)  # 信号量通信
    
    def run(self):
        while self.running:  # 多线程检测循环
            results = self.model(frame, conf=self.conf, iou=self.iou)
            self.frame_received.emit(original_frame, result_frame, detections)

七、项目源码(视频简介)

演示与介绍视频:

基于深度学习YOLOv12的猫狗品种识别检测系统(YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型)_哔哩哔哩_bilibili

基于深度学习YOLOv12的猫狗品种识别检测系统(YOLOv12+YOLO数据集+UI界面+登录注册界面+Python项目源码+模型)

数据集介绍:濒危野生动物目标检测数据集 一、基础信息 数据集名称:濒危野生动物目标检测数据集 图片数量: - 训练集:6,409张图片 - 验证集:712张图片 分类类别: - Badger(獾) - Black Bear(黑熊) - Cheetah(猎豹) - Hare(野兔) - Leopard Cat(豹猫) - Musk-Deer(麝) - Northeast Tiger(东北虎) - Northeast-Leopard(东北豹) - Red-Fox(红狐) 标注格式: YOLO格式,包含边界框和类别标签,适用于目标检测任务。 二、适用场景 生态保护与监测: 支持野生动物保护机构开发AI监测系统,用于自然保护区、森林等场景中濒危物种的实时识别与追踪。 农业与畜牧业防护: 集成至农田监控系统检测可能破坏作物的野生动物(如野兔、獾),帮助农户采取防护措施。 自动驾驶环境感知: 用于训练自动驾驶模型识别道路周边野生动物(如黑熊、狐狸),提升行车安全预警能力。 学术研究与教育: 为生物学、生态学领域提供数据支撑,支持濒危动物行为研究与保护策略制定。 三、数据集优势 物种多样性覆盖: 包含9种典型野生动物,涵盖食肉目、偶蹄目等多个生态位,尤其聚焦东北虎、东北豹等濒危物种。 实际场景适配性: 数据来源包含不同光照与环境条件(如夜间黑熊样本),强化模型在复杂自然场景中的鲁棒性。 标注精准度高: YOLO格式标注经过多轮校验,边界框定位准确,可直接用于主流目标检测框架训练。 保护价值突出: 数据涵盖多类国家一级/二级保护动物,为AI技术在生物多样性保护中的落地提供关键基础设施。
准备好深入研究可爱的猫咪分类了吗?这个数据集揭示了超过11,000张高质量的图片,涵盖了60多种不同的猫品种。从波斯猫到优雅的暹罗猫,从蓬松的缅因州猫到好奇的无毛猫,你将发现我们这些完美的毛茸茸伙伴的惊艳视觉百科全书 品种: 暹罗猫 波斯猫 缅因猫 布偶猫 无毛猫(斯芬克斯猫) 英国短毛猫 阿比西尼亚猫 苏格兰折耳猫 德文卷毛猫 美国短毛猫 东方短毛猫 缅甸猫 异国短毛猫(加菲猫) 挪威森林猫 西伯利亚猫 土耳其安哥拉猫 孟加拉豹猫 缅甸圣猫(伯曼猫) 康沃尔卷毛猫 埃及猫(埃及毛) 日本短尾猫 科拉特猫 拉波姆猫 曼岛猫(无尾猫) 奥西猫 俄罗斯蓝猫 萨凡纳猫 赛尔凯克卷毛猫 新加坡猫 索马里猫 东奇尼猫 土耳其凡湖猫 美国短尾猫 美国卷耳猫 美国刚毛猫 巴厘猫 孟买猫 夏特尔猫 金吉拉长毛猫(辛布里克猫) 欧洲短毛猫 哈瓦那棕猫 喜马拉雅猫 考马尼猫 韩国短尾猫 狼人猫(莱科伊猫) 矮脚猫(曼基康猫) 内布鲁猫 彼得秃猫 精灵猫(皮克西猫) 布偶亲缘猫(拉格多芬猫) 塞伦盖蒂猫 泰国猫 玩具虎猫(托伊格猫) 乌克兰勒夫科伊猫 雪橇猫(乔西猫) 塞浦路斯猫 顿斯科伊猫 德国卷毛猫 卡累利阿短尾猫 库里连短尾猫 湄公河短尾猫 俄勒冈卷毛猫 狩猎猫(Safari猫) 索科克猫 乌拉尔卷毛猫 凡卡迪斯猫(Vankedisi,凡湖猫别名)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值