GitHub - priyagillan52/YOLOEncoderUNetDecoder: ML/DL final project for object segmentation. We used a YOLOv5 model as our encoder and created a custom UNet for our decoder

priyagillan52 / YOLOEncoderUNetDecoder Public

ML/DL final project for object segmentation. We used a YOLOv5 model as our encoder and created a custom UNet for our decoder

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
data		data
runs/detect		runs/detect
DLFinalv1.ipynb		DLFinalv1.ipynb
DLFinalv2.ipynb		DLFinalv2.ipynb
DLFinalv3.ipynb		DLFinalv3.ipynb
mymodel.pth		mymodel.pth
yolov5s.pt		yolov5s.pt
yolov8n.pt		yolov8n.pt