kebincontreras commited on
Commit
44ed026
·
verified ·
1 Parent(s): f1922f2

Delete notacion.py

Browse files
Files changed (1) hide show
  1. notacion.py +0 -30
notacion.py DELETED
@@ -1,30 +0,0 @@
1
- import numpy as np
2
- from PIL import Image
3
- import torch
4
- from torchvision import transforms
5
-
6
- # Mapeo de clases de YOLOv10 a las clases de tu base de datos
7
-
8
-
9
- def map_yolo_classes_to_db(yolo_detections):
10
- """
11
- Ajusta las clases detectadas por YOLO a las clases de la base de datos según el mapeo proporcionado.
12
- """
13
- mapping = {
14
- 'person': 'pedestrian',
15
- 'bicycle': 'cyclist',
16
- 'car': 'car',
17
- 'motorbike': 'cisc',
18
- 'aeroplane': 'cisc',
19
- 'bus': 'cisc',
20
- 'train': 'tram',
21
- 'truck': 'truck',
22
- 'van': 'van'
23
- }
24
- mapped_detections = []
25
- for det in yolo_detections:
26
- if det['class'] in mapping:
27
- det['class'] = mapping[det['class']]
28
- mapped_detections.append(det)
29
- return mapped_detections
30
-