Skip to content

Latest commit

 

History

History
26 lines (20 loc) · 1.6 KB

INFO.md

File metadata and controls

26 lines (20 loc) · 1.6 KB

About

Handwritten digits detection using a YOLOv8 detection model and ONNX pre/post processing. An example of how model works in real world scenario can be viewed at https://thawro.github.io/web-object-detector/.

Data

The dataset consists of images created with the use of a HWD+ dataset (more here).

Pipeline

Each pipeline step is done with ONNX models. The complete pipeline during inference is the following:

  1. Image preprocessing - resize and pad to match model input size (preprocessing)
  2. Object detection - Detect objects with YOLOv8 model (yolo)
  3. Non Maximum Supression - Apply NMS to YOLO output (nms)
  4. Postprocessing - Apply postprocessing to filtered boxes (postprocessing)

Tech stack

  • PyTorch - neural networks architectures and datasets classes
  • ONNX - All processing steps used in pipeline
  • ONNX Runtime - Pipeline inference
  • OpenCV - Image processing for the server-side model inference (optional)
  • React - Web application used to test object detection models in real world examples

App instruction

  1. Go to https://thawro.github.io/web-object-detector/
  2. Follow the instructions on the page