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

Group R-CNN for Point-based Weakly Semi-supervised Object Detection (CVPR2022)

Geometric Vector Perceptron --- a rotation-equivariant GNN for learning from biomolecular structure

Catbird is an open source paraphrase generation toolkit based on PyTorch.

Implementation of Kalman Filter in Python

Code for ICMI2020 and ICMI2021 papers: "Studying Person-Specific Pointing and Gaze Behavior for Multimodal Referencing of Outside Objects from a Moving Vehicle" and "ML-PersRef: A Machine Learning-based Personalized Multimodal Fusion Approach for Referencing Outside Objects From a Moving Vehicle"

MobileNetV1-V2，MobileNeXt，GhostNet，AdderNet，ShuffleNetV1-V2，Mobile+ViT etc.

Transport Mode detection - can detect the mode of transport with the help of features such as acceeration,jerk etc

Code for Learning Manifold Patch-Based Representations of Man-Made Shapes, in ICLR 2021.

Official Implementation of "LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks"

Automatic library of congress classification, using word embeddings from book titles and synopses.

Visual Memorability for Robotic Interestingness via Unsupervised Online Learning (ECCV 2020 Oral and TRO)

Jupyter Dock is a set of Jupyter Notebooks for performing molecular docking protocols interactively, as well as visualizing, converting file formats and analyzing the results.

GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training @ KDD 2020

Nicely is a real-time Feedback and Intervention Program Depression is a prevalent issue across all age groups, socioeconomic classes, and cultural identities.

Multi-objective constrained optimization for energy applications via tree ensembles

OSLO: Open Source framework for Large-scale transformer Optimization

Tool which allow you to detect and translate text.

Python library for loading and using triangular meshes.

NeuralTalk is a Python+numpy project for learning Multimodal Recurrent Neural Networks that describe images with sentences.

IEEE Winter Conference on Applications of Computer Vision 2022 Accepted