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

Wafer Fault Detection using MlOps Integration

Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)

Implementation of UNet on the Joey ML framework

Tensorflow solution of NER task Using BiLSTM-CRF model with Google BERT Fine-tuning And private Server services

State-Relabeling Adversarial Active Learning

Largest list of models for Core ML (for iOS 11+)

VolumeGAN - 3D-aware Image Synthesis via Learning Structural and Textural Representations

Fast and accurate optimisation for registration with little learningconvexadam

Ludwig is a toolbox that allows to train and evaluate deep learning models without the need to write code.

Data Engineering ZoomCamp

Rule based classification A hotel s customers dataset

Adaptive, interpretable wavelets across domains (NeurIPS 2021)

Neural Magic Eye: Learning to See and Understand the Scene Behind an Autostereogram, arXiv:2012.15692.

PyTorch implementation of NeurIPS 2021 paper: "CoFiNet: Reliable Coarse-to-fine Correspondences for Robust Point Cloud Registration"

Fake-user-agent-traffic-geneator - Python CLI Tool to generate fake traffic against URLs with configurable user-agents

Code for MSc Quantitative Finance Dissertation

Efficient Deep Learning Systems course

Faster RCNN pytorch windows

Official code for On Path Integration of Grid Cells: Group Representation and Isotropic Scaling (NeurIPS 2021)

[CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers