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

A Kernel fuzzer focusing on race bugs

This repository contains the code for "SBEVNet: End-to-End Deep Stereo Layout Estimation" paper by Divam Gupta, Wei Pu, Trenton Tabor, Jeff Schneider

Underwater industrial application yolov5m6

[ICRA2021] Reconstructing Interactive 3D Scene by Panoptic Mapping and CAD Model Alignment

Group Fisher Pruning for Practical Network Compression(ICML2021)

i3DMM: Deep Implicit 3D Morphable Model of Human Heads

A Python library for common tasks on 3D point clouds

Automate issue discovery for your projects against Lightning nightly and releases.

for taichi voxel-challange event

HDMapNet: A Local Semantic Map Learning and Evaluation Framework

Repositorio oficial del curso IIC2233 Programación Avanzada 🚀✨

An experiment on the performance of homemade Q-learning AIs in Agar.io depending on their state representation and available actions

A High-Performance Distributed Library for Large-Scale Bundle Adjustment

Semi-supervised Semantic Segmentation with Directional Context-aware Consistency (CVPR 2021)

Research into Forex price prediction from price history using Deep Sequence Modeling with Stacked LSTMs.

DeepOBS: A Deep Learning Optimizer Benchmark Suite

Optimizing DR with hard negatives and achieving SOTA first-stage retrieval performance on TREC DL Track (SIGIR 2021 Full Paper).

Flybirds - BDD-driven natural language automated testing framework, present by Trip Flight

CTC segmentation python package

Garbage Detection system which will detect objects based on whether it is plastic waste or plastics or just garbage.