Detekce osob na dopravním prostředku pomocí YOLO algoritmu

Ondřej Binar

Bakalářská práce

Person Detection on the Vehicle Using the YOLO Algorithm

Anotace:

Bakalářská práce se zabývá problematikou detekce osob na dopravním prostředku pomocí YOLO algoritmu. V úvodu jsou vysvětleny základy strojového a hlubokého učení, druhy strojového rozpoznávání, druhy algoritmů pro detekci, historie, princip a funkce YOLO algoritmu včetně porovnání jednotlivých verzí. Na základě zjištěných informací jsou vytvořeny potřebné datové sady a programy pro naučení vlastního detekčního modelu pro detekci osob na dopravním prostředku. Naučený detekční model je verifikován na testovacím video souboru. Na základě verifikace je následně vytvořená datová sada rozšířena o koláže a jsou trénovány další detekční modely, které jsou porovnávány mezi sebou. V závěru vlastní práce je hodnocení detekčních modelů a popis sledu událostí pro možnost trénování vlastního detekčního modelu.

Abstract:

The bachelor thesis deals with the problem of detection of people on the vehicle using the YOLO algorithm. In the introduction, the basics of machine and deep learning, types of machine recognition, types of detection algorithms, history, principle and function of YOLO algorithm including comparison of different versions are explained. Based on the information, the necessary datasets and programs are developed to learn a custom detection model for detecting people on a vehicle. The learned detection model is verified on a test video file. Based on the verification, the created dataset is then augmented with collages and other detection models are trained and compared with each other. The actual work concludes with an evaluation of the detection models and a description of the sequence of events to enable the training of a custom detection model.

Klíčová slova

detekce objektů, YOLO, dopravní prostředky, AI, strojové učení

Keywords

object detection, YOLO, vehicles, AI, machine learning

Jazyk práce: čeština

Datum vytvoření / odevzdání či podání práce: 13. 5. 2024

Obhajoba závěrečné práce

• Obhajoba proběhla 6. 6. 2024
• Vedoucí: Tomáš Spurný
• Oponent: Zdeněk Zeman

VŠB – Technická univerzita Ostrava

Fakulta strojní

Bakalářský studijní program:
Mechatronika