aagamjtdev commited on
Commit
7d7e97e
·
1 Parent(s): 9ae0a89

YOLO weights

Browse files
YOLO_MATH/yolo_split_data/data.yaml ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # data.yaml (Save this next to the 'yolo_split_data' folder)
2
+
3
+ # Root path where the split data is located
4
+ path: /home/dipesh/Downloads/api-mcq/YOLO_MATH/yolo_split_data
5
+
6
+ # Paths to the split directories (relative to the 'path' above)
7
+ train: images/train
8
+ val: images/val
9
+ test: images/test
10
+
11
+ # Number of classes
12
+ nc: 2
13
+
14
+ # Class names
15
+ names: ['equation', 'figure']
YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/args.yaml ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ task: detect
2
+ mode: train
3
+ model: ./runs/detect/math_figure_detector_final2/weights/best.pt
4
+ data: ./data.yaml
5
+ epochs: 50
6
+ time: null
7
+ patience: 50
8
+ batch: 8
9
+ imgsz: 640
10
+ save: true
11
+ save_period: -1
12
+ cache: false
13
+ device: cpu
14
+ workers: 8
15
+ project: runs/detect
16
+ name: math_figure_detector_v3
17
+ exist_ok: false
18
+ pretrained: true
19
+ optimizer: auto
20
+ verbose: true
21
+ seed: 0
22
+ deterministic: true
23
+ single_cls: false
24
+ rect: false
25
+ cos_lr: false
26
+ close_mosaic: 10
27
+ resume: false
28
+ amp: true
29
+ fraction: 1.0
30
+ profile: false
31
+ freeze: null
32
+ multi_scale: false
33
+ compile: false
34
+ overlap_mask: true
35
+ mask_ratio: 4
36
+ dropout: 0.0
37
+ val: true
38
+ split: val
39
+ save_json: false
40
+ conf: null
41
+ iou: 0.7
42
+ max_det: 300
43
+ half: false
44
+ dnn: false
45
+ plots: true
46
+ source: null
47
+ vid_stride: 1
48
+ stream_buffer: false
49
+ visualize: false
50
+ augment: false
51
+ agnostic_nms: false
52
+ classes: null
53
+ retina_masks: false
54
+ embed: null
55
+ show: false
56
+ save_frames: false
57
+ save_txt: false
58
+ save_conf: false
59
+ save_crop: false
60
+ show_labels: true
61
+ show_conf: true
62
+ show_boxes: true
63
+ line_width: null
64
+ format: torchscript
65
+ keras: false
66
+ optimize: false
67
+ int8: false
68
+ dynamic: false
69
+ simplify: true
70
+ opset: null
71
+ workspace: null
72
+ nms: false
73
+ lr0: 0.01
74
+ lrf: 0.01
75
+ momentum: 0.937
76
+ weight_decay: 0.0005
77
+ warmup_epochs: 3.0
78
+ warmup_momentum: 0.8
79
+ warmup_bias_lr: 0.1
80
+ box: 7.5
81
+ cls: 0.5
82
+ dfl: 1.5
83
+ pose: 12.0
84
+ kobj: 1.0
85
+ nbs: 64
86
+ hsv_h: 0.015
87
+ hsv_s: 0.7
88
+ hsv_v: 0.4
89
+ degrees: 5.0
90
+ translate: 0.1
91
+ scale: 0.5
92
+ shear: 0.0
93
+ perspective: 0.0
94
+ flipud: 0.5
95
+ fliplr: 0.5
96
+ bgr: 0.0
97
+ mosaic: 1.0
98
+ mixup: 0.0
99
+ cutmix: 0.0
100
+ copy_paste: 0.0
101
+ copy_paste_mode: flip
102
+ auto_augment: randaugment
103
+ erasing: 0.4
104
+ cfg: null
105
+ tracker: botsort.yaml
106
+ save_dir: /home/dipesh/Downloads/api-mcq/YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3
YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/best.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30e67325439ec2068de08eda75160a2228bce4bee4ddf7c243f8ebb09b75d327
3
+ size 6235171
YOLO_MATH/yolo_split_data/runs/detect/math_figure_detector_v3/weights/last.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:312749aad61dad27ee200320f5d83ca87eea859c473902acf363e95c50790488
3
+ size 6235171
packages.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+
2
+ tesseract-ocr
3
+ tesseract-ocr-eng
4
+ libglib2.0-0
requirements.txt CHANGED
@@ -5,8 +5,9 @@ gradio==4.44.0
5
  torch==2.1.0+cpu
6
  --extra-index-url https://download.pytorch.org/whl/cpu
7
 
8
- # Transformers and NLP
9
- transformers==4.35.0
 
10
  TorchCRF==1.1.0
11
 
12
  # Computer Vision
@@ -25,6 +26,3 @@ Pillow==10.1.0
25
  # Scientific Computing
26
  numpy==1.24.3
27
  scipy==1.11.4
28
-
29
- # Utilities - Remove strict version pin to let pip resolve
30
- huggingface-hub>=0.16.4,<1.0
 
5
  torch==2.1.0+cpu
6
  --extra-index-url https://download.pytorch.org/whl/cpu
7
 
8
+ # Transformers and NLP - Updated versions for compatibility
9
+ transformers==4.37.0
10
+ tokenizers>=0.15.0
11
  TorchCRF==1.1.0
12
 
13
  # Computer Vision
 
26
  # Scientific Computing
27
  numpy==1.24.3
28
  scipy==1.11.4