Data

Tools

Indexes

Applications

Experiments

Open Source Science

yolov5-classify

yolov5 for classify and detection

https://github.com/cdmstrong/yolov5-classify

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 (0.7%) to scientific vocabulary
Last synced: 6 months ago · JSON representation ·

Repository

yolov5 for classify and detection

Basic Info
  • Host: GitHub
  • Owner: cdmstrong
  • License: agpl-3.0
  • Language: Python
  • Default Branch: main
  • Size: 22.5 MB
Statistics
  • Stars: 2
  • Watchers: 1
  • Forks: 0
  • Open Issues: 5
  • Releases: 0
Created over 2 years ago · Last pushed over 2 years ago
Metadata Files
Readme Contributing License Citation

README.md

'''训练'''： python classify/train.py --model weights/yolov5s-cls.pt --data data/classify/data_patch --epochs 100 --imgsz 64 --device 1

训练数据分成如下

data_patch ---train --- class1 --- class2 --- class3 ---val --- class1 --- class2 --- class3

网络结构

特征提取使用的是 v5的特征提取 最后加上 classify 自定义的 分类网络

使用配置文件yaml 生成网络

默认配置文件为 - yolov5s-cls.yaml

修改了原有的v5 结构，保留backbone部分， 在head 新增Classify 层

使用预训练模型生成分类模型

原理： 提取模型的backbone部分， 替换head部分为分类层

Owner

  • Name: 东哥爱编程
  • Login: cdmstrong
  • 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
Last Year

Dependencies

.github/workflows/ci-testing.yml actions
  • actions/checkout v3 composite
  • actions/setup-python v4 composite
  • slackapi/slack-github-action v1.23.0 composite
.github/workflows/codeql-analysis.yml actions
  • actions/checkout v3 composite
  • github/codeql-action/analyze v2 composite
  • github/codeql-action/autobuild v2 composite
  • github/codeql-action/init v2 composite
.github/workflows/docker.yml actions
  • actions/checkout v3 composite
  • docker/build-push-action v4 composite
  • docker/login-action v2 composite
  • docker/setup-buildx-action v2 composite
  • docker/setup-qemu-action v2 composite
.github/workflows/greetings.yml actions
  • actions/first-interaction v1 composite
.github/workflows/links.yml actions
  • actions/checkout v3 composite
  • nick-invision/retry v2 composite
.github/workflows/stale.yml actions
  • actions/stale v8 composite
.github/workflows/translate-readme.yml actions
  • actions/checkout v3 composite
  • actions/setup-node v3 composite
  • dephraiim/translate-readme main 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
requirements.txt pypi
  • Pillow >=7.1.2
  • PyYAML >=5.3.1
  • gitpython >=3.1.30
  • matplotlib >=3.3
  • numpy >=1.18.5
  • opencv-python >=4.1.1
  • pandas >=1.1.4
  • psutil *
  • requests >=2.23.0
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • setuptools >=65.5.1
  • thop >=0.1.1
  • torchvision >=0.8.1
  • tqdm >=4.64.0
utils/google_app_engine/additional_requirements.txt pypi
  • Flask ==2.3.2
  • gunicorn ==19.10.0
  • pip ==21.1
  • werkzeug >=2.2.3