/
efr
/
SmartCity
Обзор
Документация
Войти
/
efr
/
SmartCity
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
utils/scale_object.py
19 строк
753 B
Sergey
add_project
27 фев 2025, 19:33
27 фев 2025, 19:33
7eeda08
Код
Авторство
О чём код?
import settings def scale(results, orig_width, orig_height): scaled_objects = [] for result in results: for box in result.boxes.cpu().numpy(): x1,y1,x2,y2 = box.xyxy[0].astype(int) class_id = int(box.cls[0]) if class_id not in settings.target_classes: continue scaled_objects.append({ "coords": (int(x1*(orig_width/640)), int(y1*(orig_height/480)), int(x2*(orig_width/640)), int(y2*(orig_height/480))), "class_id": class_id, "track_id": int(box.id[0]) if box.id is not None else None, "label": settings.target_classes[class_id] }) return scaled_objects