How do you add a class to coco classes in YOLO (CNN) object detection?

This page summarizes the projects mentioned and recommended in the original post on /r/computervision

Our great sponsors
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • WorkOS - The modern identity platform for B2B SaaS
  • SaaSHub - Software Alternatives and Reviews
  • darknet

    YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet ) (by AlexeyAB)

  • The instructions in the AlexeyAB repo are quite straightforward. You can follow along here. I would recommend adding your class as the last line in coco.names and then only include your new class in the retraining data. This will be much faster retraining with all of the Coco data but you must be careful that your new data does not include unlabelled instances of any of the other 80 classes or else performance in the other classes may be reduced.

  • ScaledYOLOv4

    Scaled-YOLOv4: Scaling Cross Stage Partial Network

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo
NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts