|
|
import numpy as np
|
|
|
from PIL import Image
|
|
|
import torch
|
|
|
from torchvision import transforms
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def map_yolo_classes_to_db(yolo_detections):
|
|
|
"""
|
|
|
Ajusta las clases detectadas por YOLO a las clases de la base de datos según el mapeo proporcionado.
|
|
|
"""
|
|
|
mapping = {
|
|
|
'person': 'pedestrian',
|
|
|
'bicycle': 'cyclist',
|
|
|
'car': 'car',
|
|
|
'motorbike': 'cisc',
|
|
|
'aeroplane': 'cisc',
|
|
|
'bus': 'cisc',
|
|
|
'train': 'tram',
|
|
|
'truck': 'truck',
|
|
|
'van': 'van'
|
|
|
}
|
|
|
mapped_detections = []
|
|
|
for det in yolo_detections:
|
|
|
if det['class'] in mapping:
|
|
|
det['class'] = mapping[det['class']]
|
|
|
mapped_detections.append(det)
|
|
|
return mapped_detections
|
|
|
|
|
|
|
