Cél: Szemdetektálás képeken, fine-tuningolás, nem egy tökéletesen működő modell betanítása, hanem a fine-tuningolás megértése.
Tartalom: A dataset (annotált képek), YOLO11n modell, data.yaml fájl, és egy Jupyter jegyzetfüzet, ami a dokumentációt is tartalmazza.
Követelmények
- Python: 3.8+
- Függőségek:
Ultralytics(YOLOv8),PyTorch(CUDA 12.8-as verzió)
Megjegyzés
A teszt_kepek/ könyvtárban szereplő egyes képek AI-generáltak, ezeken a képeken a személyek nem léteznek.
A projekt részletesebb dokumentációja megtalálható a projekt.ipynb jegyzetfüzeten belül!
Források
- Dataset: https://www.kaggle.com/datasets/icebearogo/eye-detection-dataset
- YOLO11n modell: https://docs.ultralytics.com/models/yolo11/#detection-coco
- Teszt képek: https://thispersondoesnotexist.com/