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

Information-Theoretic Multi-Objective Bayesian Optimization with Continuous Approximations

Python package provinding tools for artistic interactive applications using AI

Release of the ConditionalQA dataset

[KDD 2021, Research Track] DiffMG: Differentiable Meta Graph Search for Heterogeneous Graph Neural Networks

Code release for Hu et al. Segmentation from Natural Language Expressions. in ECCV, 2016

Cervix ROI Segmentation Using U-NET

Website for D2C paper

PyTorch implementation of the end-to-end coreference resolution model with different higher-order inference methods.

Code for the paper "Adapting Monolingual Models: Data can be Scarce when Language Similarity is High"

A video scene detection algorithm is designed to detect a variety of different scenes within a video

AnimationKit: AI Upscaling & Interpolation using Real-ESRGAN+RIFE

Photographic Image Synthesis with Cascaded Refinement Networks - Pytorch Implementation

A stock generator that assess a list of stocks and returns the best stocks for investing and money allocations based on users choices of volatility, duration and number of stocks

Simple converter for deploying Stable-Baselines3 model to TFLite and/or Coral

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Code of Periodic Activation Functions Induce Stationarity

S2s2net - Sentinel-2 Super-Resolution Segmentation Network

PyTorch implementation of DCT fast weight RNNs

The first machine learning framework that encourages learning ML concepts instead of memorizing class functions.

ServiceX Transformer that converts flat ROOT ntuples into columnwise data