Automatická detekce odpadu v obrazových datech

Abstract

Cílem diplomové práce je vytvořit software pro automatickou detekci odpadu v prostředí Python pomocí konvolučních neuronových sítí. Teoretická část je věnována problematice odpadu a recyklace. Práce popisuje strukturu konvoluční neuronové sítě a její učení. Dále se zabývá algoritmy a metriky pro klasifikaci a detekci odpadu. Praktická část obsahuje dataset se shromážděnými daty a jejich augmentace, které jsou využity k následnému trénování a optimalizaci modelu. V práci je zvolen algoritmus YOLOv5 a jeho modifikace YOLOv5n, YOLOv5s a YOLOv5m, na kterých je síť postupně trénována a následně dle zvolených metrik vyhodnocena. Součástí je také jednoduchá aplikace, pomocí které lze jednoduše vyhodnotit snímky.