leeyunjai commited on
Commit
37df54c
1 Parent(s): 580cc71

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +53 -0
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Number and Operator Detection Based on YOLO11x
2
+
3
+ This repository contains a PyTorch-exported model for detecting numbers (0-10) and basic arithmetic operators (`+`, `-`, `×`, `÷`, `=`) using the YOLO11x architecture. The model has been trained to recognize these symbols in images and return their locations and classifications.
4
+
5
+ ## Model Description
6
+
7
+ The YOLO11x model is optimized for detecting the following:
8
+
9
+ - **Numbers**: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10
10
+ - **Operators**: `+`, `-`, `×`, `÷`, `=`
11
+
12
+ This model is ideal for applications involving handwritten equations, OCR tasks, or visual recognition of mathematical symbols in real-world images.
13
+
14
+ ## How to Use
15
+
16
+ To use this model in your project, follow the steps below:
17
+
18
+ ### 1. Installation
19
+
20
+ Ensure you have the `ultralytics` library installed, which is used for YOLO models:
21
+
22
+ ```bash
23
+ pip install ultralytics
24
+ ```
25
+
26
+ ### 2. Load the Model
27
+
28
+ You can load the model and perform detection on an image as follows:
29
+ ```python
30
+ from ultralytics import YOLO
31
+
32
+ # Load the model
33
+ model = YOLO("./number_11x.pt")
34
+
35
+ # Perform detection on an image
36
+ results = model("image.png")
37
+
38
+ # Display or process the results
39
+ results.show() # This will display the image with detected objects
40
+ ```
41
+
42
+ ### 3. Model Inference
43
+ The results object contains bounding boxes, labels (e.g., numbers or operators), and confidence scores for each detected object.
44
+
45
+ Access them like this:
46
+
47
+ ```python
48
+ for result in results:
49
+ print(result.boxes) # Bounding boxes
50
+ print(result.names) # Detected classes
51
+ print(result.scores) # Confidence scores
52
+ ```
53
+