Data

Tools

Indexes

Applications

Experiments

Open Source Science

yolov5-cuda12_6-training

https://github.com/chengyull/yolov5-cuda12_6-training

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ✓
    CITATION.cff file
    Found CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ○
    Academic publication links
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (1.5%) to scientific vocabulary
Last synced: 6 months ago · JSON representation ·

Repository

Basic Info
  • Host: GitHub
  • Owner: ChengYull
  • License: agpl-3.0
  • Language: Python
  • Default Branch: master
  • Size: 51.2 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 1
  • Releases: 0
Created 10 months ago · Last pushed 10 months ago
Metadata Files
Readme Contributing License Citation

README.md

YOLOv5 Custom Training with CUDA 12.6 (RTX 2050)

基于 CUDA 12.6 和 RTX 2050 的 YOLOv5 自定义目标检测训练

训练结果示例：检测自定义目标（Doro 头像）

项目简介

本项目使用 YOLOv5 在 CUDA 12.6 + RTX 2050 环境下训练自定义目标检测模型，包含： - 完整环境配置指南（Anaconda + CUDA + PyTorch） - 数据集标注教程（LabelImg 工具） - YOLOv5 模型训练代码（支持自定义数据） - 模型测试脚本（图片/视频/摄像头实时检测）

硬件要求

  • NVIDIA GPU（支持 CUDA 12.6+，如 RTX 2050/3060 等）

部署

安装依赖

bash pip install -r requirements.txt

数据集准备

  • 图片标注
  • 使用 LabelImg 工具进行数据集标注，生成 YOLO 格式的标签文件。
    • 数据集目录结构示例： train/ ├── images/ # 存放图片 ├── labels/ # 存放标注文件 └── doro.yaml # 数据集配置文件

训练

执行训练脚本：src/test_train.py

测试

执行测试脚本：src/test_doro.py

详细教程

博客园地址：https://www.cnblogs.com/1873cy/p/18844467

Owner

  • Login: ChengYull
  • Kind: user

Citation (CITATION.cff) 

cff-version: 1.2.0
preferred-citation:
  type: software
  message: If you use YOLOv5, please cite it as below.
  authors:
  - family-names: Jocher
    given-names: Glenn
    orcid: "https://orcid.org/0000-0001-5950-6979"
  title: "YOLOv5 by Ultralytics"
  version: 7.0
  doi: 10.5281/zenodo.3908559
  date-released: 2020-5-29
  license: AGPL-3.0
  url: "https://github.com/ultralytics/yolov5"

GitHub Events

Total
  • Watch event: 1
  • Issue comment event: 3
  • Push event: 1
  • Pull request event: 1
  • Create event: 4
Last Year
  • Watch event: 1
  • Issue comment event: 3
  • Push event: 1
  • Pull request event: 1
  • Create event: 4

Dependencies

.github/workflows/ci-testing.yml actions
  • actions/checkout v4 composite
  • actions/setup-python v5 composite
  • slackapi/slack-github-action v2.0.0 composite
.github/workflows/cla.yml actions
  • contributor-assistant/github-action v2.6.1 composite
.github/workflows/docker.yml actions
  • actions/checkout v4 composite
  • docker/build-push-action v6 composite
  • docker/login-action v3 composite
  • docker/setup-buildx-action v3 composite
  • docker/setup-qemu-action v3 composite
.github/workflows/format.yml actions
  • ultralytics/actions main composite
.github/workflows/links.yml actions
  • actions/checkout v4 composite
  • ultralytics/actions/retry main composite
.github/workflows/merge-main-into-prs.yml actions
  • actions/checkout v4 composite
  • actions/setup-python v5 composite
.github/workflows/stale.yml actions
  • actions/stale v9 composite
utils/docker/Dockerfile docker
  • pytorch/pytorch 2.0.0-cuda11.7-cudnn8-runtime build
utils/google_app_engine/Dockerfile docker
  • gcr.io/google-appengine/python latest build
pyproject.toml pypi
  • matplotlib >=3.3.0
  • numpy >=1.22.2
  • opencv-python >=4.6.0
  • pandas >=1.1.4
  • pillow >=7.1.2
  • psutil *
  • py-cpuinfo *
  • pyyaml >=5.3.1
  • requests >=2.23.0
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • thop >=0.1.1
  • torch >=1.8.0
  • torchvision >=0.9.0
  • tqdm >=4.64.0
  • ultralytics >=8.1.47
requirements.txt pypi
  • PyYAML >=5.3.1
  • gitpython >=3.1.30
  • matplotlib >=3.3
  • numpy >=1.23.5
  • opencv-python >=4.1.1
  • pandas >=1.1.4
  • pillow >=10.3.0
  • psutil *
  • requests >=2.32.2
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • setuptools >=70.0.0
  • thop >=0.1.1
  • torchvision >=0.9.0
  • tqdm >=4.66.3
utils/google_app_engine/additional_requirements.txt pypi
  • Flask ==2.3.2
  • gunicorn ==23.0.0
  • pip ==23.3
  • werkzeug >=3.0.1
  • zipp >=3.19.1