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

A large-scale video dataset for the training and evaluation of 3D human pose estimation models

SingleVC performs any-to-one VC, which is an important component of MediumVC project.

Generative Autoregressive, Normalized Flows, VAEs, Score-based models (GANVAS)

3D Human Pose Machines with Self-supervised Learning

MPI Interest Group on Algorithms on 1st semester 2021

Implementation of "DeepOrder: Deep Learning for Test Case Prioritization in Continuous Integration Testing".

Python scripts form performing stereo depth estimation using the high res stereo model in PyTorch .

AI Virtual Calculator: This is a simple virtual calculator based on Artificial intelligence.

PyTorch implementation of Spiking Neural Networks trained on surrogate gradient & BPTT using snntorch.

Public repository of the 3DV 2021 paper "Generative Zero-Shot Learning for Semantic Segmentation of 3D Point Clouds"

CaLiGraph Ontology as a Challenge for Semantic Reasoners ([email protected]'21)

Code for paper [ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot] (ICCV 2021, oral))

Edge Restoration Quality Assessment

Learning based AI for playing multi-round Koi-Koi hanafuda card games. Have fun.

SOTA easy to use PyTorch-based DL training library

Unofficial implementation of Pix2SEQ

OpenCVのGrabCut()を利用したセマンティックセグメンテーション向けアノテーションツール(Annotation tool using GrabCut() of OpenCV. It can be used to create datasets for semantic segmentation.)

Symbolic Parallel Adaptive Importance Sampling for Probabilistic Program Analysis in JAX

Deep Learning GPU Training System

A machine learning package for streaming data in Python. The other ancestor of River.