Deep Learning project for waste object detection and segmentation using YOLOv8 and U-Net on the TACO dataset. Implements object detection with YOLOv8 for bounding boxes and custom U-Net for pixel-wise segmentation. Includes EDA, data augmentation pipeline, and comparative analysis. Built with PyTorch for CS4045 course project.
-
Notifications
You must be signed in to change notification settings - Fork 0
Deep Learning project for waste object detection and segmentation using YOLOv8 and U-Net on the TACO dataset. Implements object detection with YOLOv8 for bounding boxes and custom U-Net for pixel-wise segmentation. Includes EDA, data augmentation pipeline, and comparative analysis. Built with PyTorch for CS4045 course project.
License
Umer-Farooq-CS/Waste-Object-Detection-Segmentation
About
Deep Learning project for waste object detection and segmentation using YOLOv8 and U-Net on the TACO dataset. Implements object detection with YOLOv8 for bounding boxes and custom U-Net for pixel-wise segmentation. Includes EDA, data augmentation pipeline, and comparative analysis. Built with PyTorch for CS4045 course project.