Classifying recyclable garbage on the street with CNN
- Run the code at Step 1: Install Requirements
- Then run the codes at Step 2: Assemble our dataset. The dataset I have prepared through Roboflow will be downloaded in the file's folder.
- Once 'datasets' folder is formed, follow the 'Street-trash-3' folder and replace 'data.yaml' file with 'street_trash.yaml' file given in this page.
- Run the code at Step 3: Train Our Custom YOLOv5 model. You may change the size of the images, batch size, epoch, and the weights.