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

Multi-robot collaborative exploration and mapping through Voronoi partition and DRL in unknown environment

GAN-STEM-Conv2MultiSlice - Exploring Generative Adversarial Networks for Image-to-Image Translation in STEM Simulation

Reinforcement Learning for the Blackjack

Transformer model implemented with Pytorch

QQ Browser 2021 AI Algorithm Competition Track 1 1st Place Program

Universal Adversarial Triggers for Attacking and Analyzing NLP (EMNLP 2019)

Official Implementation (PyTorch) of "Point Cloud Augmentation with Weighted Local Transformations", ICCV 2021

SMD-Nets: Stereo Mixture Density Networks

Qlib is an AI-oriented quantitative investment platform

Simple Pose: Rethinking and Improving a Bottom-up Approach for Multi-Person Pose Estimation

Dynamic Neural Representational Decoders for High-Resolution Semantic Segmentation

Like a cowsay but without cows!

An intuitive library to extract features from time series

A Python implementation of active inference for Markov Decision Processes

Bootstrapped Representation Learning on Graphs

Code for "Learning Graph Cellular Automata"

Doing the asl sign language classification on static images using graph neural networks.

A Confidence-based Iterative Solver of Depths and Surface Normals for Deep Multi-view Stereo

Adversarial Robustness Comparison of Vision Transformer and MLP-Mixer to CNNs

Efficient Householder transformation in PyTorch