Data

Tools

Indexes

Applications

Experiments

Open Source Science

yolov8_extend_class

https://github.com/atunass/yolov8_extend_class

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

Repository

Basic Info
  • Host: GitHub
  • Owner: aTunass
  • Language: Python
  • Default Branch: main
  • Size: 1.18 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Created almost 2 years ago · Last pushed almost 2 years ago
Metadata Files
Readme Citation

README.md

yolov8extendclass

Setup

``` conda create -n python==3.8 conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=11.8 -c pytorch -c nvidia conda activate

git clone https://github.com/aTunass/yolov8extendclass.git cd yolov8extendclass python -m pip install -e . ```

IDEA

Detail: You can refer to the following blog, all my code is taken from this blog - UI Summary: You have trained the yolov8 model with 80 classes previously with the COCO set, now you have new labels to add to the model. Instead of retraining the entire model with 81 classes, you can keep the original model and train a new model with frozen backbone and the output is 1 additional class. Then we take the two head classes of these two models and concat them together and go with the backbone that was initially used or frozen.

Config

Follow the blog and follow example_config.yaml for correct implementation

Training

you can use code example_training.py for training. Note, freeze the backbone before training

Predict

After the following training you get two model weights. In example_predict.py there will be code segments such as "check", "save dict", "load model" and "predict". You can try each one while commenting on the others. after going through two steps "check" and "save dict". you run both "load model" and "predict"

Export, Validate

You can follow the remaining codes from Ultralytics

Example result

Owner

  • Name: Nguyen Hoang Anh Tuan
  • Login: aTunass
  • Kind: user

Citation (CITATION.cff) 

# This CITATION.cff file was generated with https://bit.ly/cffinit

cff-version: 1.2.0
title: Ultralytics YOLO
message: >-
  If you use this software, please cite it using the
  metadata from this file.
type: software
authors:
  - given-names: Glenn
    family-names: Jocher
    affiliation: Ultralytics
    orcid: 'https://orcid.org/0000-0001-5950-6979'
  - given-names: Ayush
    family-names: Chaurasia
    affiliation: Ultralytics
    orcid: 'https://orcid.org/0000-0002-7603-6750'
  - family-names: Qiu
    given-names: Jing
    affiliation: Ultralytics
    orcid: 'https://orcid.org/0000-0003-3783-7069'
repository-code: 'https://github.com/ultralytics/ultralytics'
url: 'https://ultralytics.com'
license: AGPL-3.0
version: 8.0.0
date-released: '2023-01-10'

GitHub Events

Total
Last Year

Dependencies

pyproject.toml pypi
  • matplotlib >=3.3.0
  • 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
  • torch >=1.8.0
  • torchvision >=0.9.0
  • tqdm >=4.64.0
  • ultralytics-thop >=0.2.5