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

2021 National Underwater Robotics Vision Optics

Eff video representation - Efficient video representation through neural fields

Implementation of "Learning Multi-Granular Hypergraphs for Video-Based Person Re-Identification"

Reducing Information Bottleneck for Weakly Supervised Semantic Segmentation (NeurIPS 2021)

A Nim frontend for pytorch, aiming to be mostly auto-generated and internally using ATen.

Kernel Point Convolutions

Official tensorflow implementation for CVPR2020 paper “Learning to Cartoonize Using White-box Cartoon Representations”

Notification Triggers for Python

Video-based open-world segmentation

E-RAFT: Dense Optical Flow from Event Cameras

GPU-accelerated Image Processing library using OpenCL

A package related to building quasi-fibration symmetries

Implementation of "JOKR: Joint Keypoint Representation for Unsupervised Cross-Domain Motion Retargeting"

Predicting 10 different clothing types using Xception pre-trained model.

Crab is a ﬂexible, fast recommender engine for Python that integrates classic information ﬁltering recommendation algorithms in the world of scientiﬁc Python packages (numpy, scipy, matplotlib).

Tensorflow-Project-Template - A best practice for tensorflow project template architecture.

Computing Shapley values using VAEAC

Pretrained Cost Model for Distributed Constraint Optimization Problems

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).

The full training script for Enformer (Tensorflow Sonnet) on TPU clusters