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

Algorithm to texture 3D reconstructions from multi-view stereo images

Code and models for "Rethinking Deep Image Prior for Denoising" (ICCV 2021)

Multi agent DDPG algorithm written in Python + Pytorch

A Domain-Agnostic Benchmark for Self-Supervised Learning

Joint Gaussian Graphical Model Estimation: A Survey

Diffusion Normalizing Flow (DiffFlow) Neurips2021

🤗 Paper Style Guide

Who calls the shots? Rethinking Few-Shot Learning for Audio (WASPAA 2021)

DumpSMBShare - A script to dump files and folders remotely from a Windows SMB share

Code Repository for Liquid Time-Constant Networks (LTCs)

Pytorch reimplementation of the Mixer (MLP-Mixer: An all-MLP Architecture for Vision)

Deep-learning X-Ray Micro-CT image enhancement, pore-network modelling and continuum modelling

POPPY (Physical Optics Propagation in Python) is a Python package that simulates physical optical propagation including diffraction

Projects for AI/ML and IoT integration for games and other presented at re:Invent 2021.

Official Python implementation of the 'Sparse deconvolution'-v0.3.0

This repository contains an implementation of the Permutohedral Attention Module in Pytorch

[NeurIPS2021] Code Release of Learning Transferable Perturbations

Source code and notebooks to reproduce experiments and benchmarks on Bias Faces in the Wild (BFW).

An Unbiased Learning To Rank Algorithms (ULTRA) toolbox

Cervix ROI Segmentation Using U-NET