Annotate datasets with a semi-trained or fully trained YOLOv5 model

Last update: May 14, 2022

Overview

YOLOv5 Auto Annotator

Annotate datasets with a semi-trained or fully trained YOLOv5 model

Prerequisites

Ubuntu >=20.04
Python >=3.7

System dependencies

sudo apt install python3-dev python3-pip

Python dependencies

cycler==0.11.0
fonttools==4.29.0
kiwisolver==1.3.2
lxml==4.6.4
numpy==1.21.4
opencv-contrib-python==4.5.5.62
opencv-python==4.5.5.62
packaging==21.3
Pillow==9.0.0
pyparsing==3.0.7
python-dateutil==2.8.2
six==1.16.0
tqdm==4.62.3

Install with the following command -

pip3 install -r requirements.txt

Run the application

Execute annotate.py in the following format -

usage: annotate.py [-h] [--viewmode] [--imgdir IMGDIR] [--annodir ANNODIR] [--confThreshold CONFTHRESHOLD] [--nmsThreshold NMSTHRESHOLD] [--width WIDTH] [--height HEIGHT] [--onnx_path ONNX_PATH] [--labels_path LABELS_PATH]

optional arguments:
  -h, --help            show this help message and exit
  --viewmode            Toggle View Mode
  --imgdir IMGDIR       Directory of images
  --annodir ANNODIR     Directory of annotations
  --confThreshold CONFTHRESHOLD
                        Class confidence
  --nmsThreshold NMSTHRESHOLD
                        NMS threshold
  --width WIDTH         Width of network input
  --height HEIGHT       Height of network input
  --onnx_path ONNX_PATH
                        Path to onnx file
  --labels_path LABELS_PATH
                        Path to labels file

Example -

python3 annotate.py --imgdir /home/kn1ght/Documents/images --annodir annotations --onnx_path models/YOLOv5s/yolov5s.onnx --labels_path models/YOLOv5s/coco.names --viewmode

Annotate datasets with a semi-trained or fully trained YOLOv5 model

Related tags

Overview

YOLOv5 Auto Annotator

Prerequisites

System dependencies

Python dependencies

Run the application

Owner

Akash James

Datasets, Transforms and Models specific to Computer Vision

Range Image-based LiDAR Localization for Autonomous Vehicles Using Mesh Maps

Automatic Number Plate Recognition using Contours and Convolution Neural Networks (CNN)

STYLER: Style Factor Modeling with Rapidity and Robustness via Speech Decomposition for Expressive and Controllable Neural Text to Speech

Sequence-to-Sequence learning using PyTorch

Minimal But Practical Image Classifier Pipline Using Pytorch, Finetune on ResNet18, Got 99% Accuracy on Own Small Datasets.

The PyTorch improved version of TPAMI 2017 paper: Face Alignment in Full Pose Range: A 3D Total Solution.

A pre-trained model with multi-exit transformer architecture.

An official TensorFlow implementation of “CLCC: Contrastive Learning for Color Constancy” accepted at CVPR 2021.

A general-purpose, flexible, and easy-to-use simulator alongside an OpenAI Gym trading environment for MetaTrader 5 trading platform (Approved by OpenAI Gym)

AOT (Associating Objects with Transformers) in PyTorch

Advanced Deep Learning with TensorFlow 2 and Keras (Updated for 2nd Edition)

[NeurIPS 2021]: Are Transformers More Robust Than CNNs? (Pytorch implementation & checkpoints)

Collaborative forensic timeline analysis

CVPR2022 (Oral) - Rethinking Semantic Segmentation: A Prototype View

Make your first PR. A beginner friendly repository made specifically for open source beginners. Add any program under any language (it can be anything from a simple program to a complex data structure algorithm). Happy coding...

Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Dask, Flink and DataFlow

👨‍💻 run nanosaur in simulation with Gazebo/Ingnition

Using Convolutional Neural Networks (CNN) for Semantic Segmentation of Breast Cancer Lesions (BRCA)

Deep Learning Based Fasion Recommendation System for Ecommerce