Yolo11 Number Operator

🚀 Number and Operator Detection Based on YOLO11x

This project provides a PyTorch-exported model using the YOLO11x architecture to detect numbers (0 - 10) and basic arithmetic operators (+, -, ×, ÷, =) in images. It can accurately identify these symbols and return their locations and classifications.

✨ Features

The YOLO11x model is optimized for detecting the following:

• Numbers: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10
• Operators: +, -, ×, ÷, =

#class
0
1
2
3
4
5
6
7
8
9
div
eqv
minus
mult
plus

📦 Installation

Ensure you have the ultralytics library installed, which is used for YOLO models:

pip install ultralytics

💻 Usage Examples

Basic Usage

from ultralytics import YOLO

# Load the model
model = YOLO("./number_11x.pt")

# Perform detection on an image
results = model("image.png")

# Display or process the results
results.show()  # This will display the image with detected objects

Advanced Usage

The results object contains bounding boxes, labels (e.g., numbers or operators), and confidence scores for each detected object. Access them like this:

for result in results:
    print(result.boxes)   # Bounding boxes
    print(result.names)   # Detected classes
    print(result.scores)  # Confidence scores