lego-bricks-ml-vision 0.1.0

A package for LEGO brick detection and visualization using YOLO.

LEGO Bricks ML Vision

A Python package for detecting and visualizing LEGO bricks using YOLO and custom visualization pipelines. This project is designed to streamline the process of detecting LEGO pieces in cluttered environments and producing meaningful visualizations for analysis and presentation.

---

Features

• Object Detection: Detect LEGO bricks using YOLOv8.
• Preprocessing Tools: Scripts to preprocess images and datasets.
• Dataset Conversion: Convert LabelMe annotations to YOLO format.
• Visualization Pipelines: Create annotated grids, comparisons, and organized folders for presentations.
• Modular Design: Easy-to-use package structure for scaling or adapting the project.

---

System Requirements

• Python Version: >= 3.8
• Supported OS: Windows, macOS, Linux
• Hardware:
  • CPU support (required)
  • GPU support (optional but recommended for training YOLO models)

---

Installation

To get started with the project, clone the repository and install the package:

git clone https://github.com/MiguelDiLalla/LEGO_Bricks_ML_Vision.git
cd LEGO_Bricks_ML_Vision
pip install -e .