Detekcia Lopty

Strojové učenie

Author

Tech stack

Python Python PyTorch PyTorch Wandb Wandb

Detekcia Lopty

📜 Popis

Projekt počítačového videnia s hlbokým učením pre detekciu a lokalizáciu lôpt na obrázkoch pomocou vlastnej CNN architektúry. Systém využíva prístup s duálnymi hlavami: binárny klasifikátor na detekciu prítomnosti lopty a regresor pre presnú lokalizáciu pomocou ohraničujúceho rámčeka.

⚙️ Funkcie

  • CNN Architektúra – 4-vrstvový konvolučný základ
  • Binárna klasifikácia – Detekcia prítomnosti lopty
  • Regresia ohraničujúceho rámčeka – Presná lokalizácia s normalizovanými YOLO-formátovými súradnicami
  • Augmentácia dát – Transformácie obrázkov pre zlepšenú generalizáciu modelu