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

Cookiecutter PyTorch Lightning

Yolox-bytetrack-sample - Python sample of MOT (Multiple Object Tracking) using YOLOX and ByteTrack

Explore extreme compression for pre-trained language models

1st Solution For NeurIPS 2021 Competition on ML4CO Dual Task

Code for "Sparse Steerable Convolutions: An Efficient Learning of SE(3)-Equivariant Features for Estimation and Tracking of Object Poses in 3D Space"

Research using Cirq!

This repository implements variational graph auto encoder by Thomas Kipf.

Code to generate datasets used in "How Useful is Self-Supervised Pretraining for Visual Tasks?"

PiRank: Learning to Rank via Differentiable Sorting

TrackTech: Real-time tracking of subjects and objects on multiple cameras

PyTorch Implementations for DeeplabV3 and PSPNet

Torch implementation of SegNet and deconvolutional network

基于PaddleOCR搭建的OCR server... 离线部署用

A simplistic and efficient pure-python neural network library from Phys Whiz with CPU and GPU support.

Code for the paper One Thing One Click: A Self-Training Approach for Weakly Supervised 3D Semantic Segmentation, CVPR 2021.

bespoke tooling for offensive security's Windows Usermode Exploit Dev course (OSED)

Look Closer: Bridging Egocentric and Third-Person Views with Transformers for Robotic Manipulation

Survival analysis (SA) is a well-known statistical technique for the study of temporal events.

mmdetection version of TinyBenchmark.

Using Random Effects to Account for High-Cardinality Categorical Features and Repeated Measures in Deep Neural Networks