Find Amenities in a House

• Developed an end-to-end web application for detecting 30 household amenities using FlaskAPI and GCP
• Used YOLOv5 model architecture on collected Dataset and achieved a mAP@0.5-0.95 of 0.4
• Improved model Precision to 43% and mAP@0.5-0.95 to 0.5 using transfer learning with COCO weights

Check full write up and User App

• personal
• machine-learning
• deep learning
• computer vision
• object detection
• cloud