Before you begin, ensure you have met the following requirements:
- Python 3.11 or later
- Pip package manager
- An environment manager like venv (optional, but recommended)
python -m venv env
venv\Scripts\activate
source venv/bin/activate
pip install torch torchvision torchaudio
pip install ultralytics
- Run
yolov8_trainer.py
to train the model - The trainer loads data from the datasets folder, this can be configured in
data.yaml
file by updating thepath
to../wisdomteethdetection/datasets/<your-dataset>/