Skip to content

Latest commit

 

History

History

YOLOv8

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

YOLOv8 Waste Detection

Preprocessing:

image

Download the plastic dataset:
https://datasetninja.com/plastic-bottles

Run the preprocessing of the dataset using the 'convert_format_file.py' script.

Inspect the remaining images with the 'dataset_inspection.py' script.

Training:

image

Train the model using the 'train.py' script with the original weights 'yolov8n.pt'.

The 'dataset.yaml' file is used to define the location of the training images.

Detection:

image

Run the 'detect.py' script for object detection on an image.

Run the 'yolo_inference.py' script for object detection on an video feed.