To get started with the project, please follow the steps below:
-
Clone the repository:
git clone https://github.com/romulo-maciel/yolo-test.git
-
Create a Python virtual environment (venv):
python3 -m venv .
-
Activate the virtual environment:
source bin/activate
-
Download the dataset zip file from here and unzip it to the
datasets/archive/
folder. -
Install the required dependencies:
pip install ultralytics
-
Run the setup script:
python setup.py
-
Train the YOLO model:
yolo train model=models/yolov8n.pt data=plants.yaml epochs=50
That's it