Furniture Object Detection using SSD | PyTorch Custom Training
🪑 Furniture Object Detection using SSD (PyTorch)
In this project, we train a custom SSD (Single Shot Detector) model with a VGG16 backbone to detect furniture objects like Chairs, Sofas, and Tables. The dataset is downloaded from Roboflow, converted into the correct format, and used to build a custom PyTorch dataset loader.
We fine-tune a pretrained SSD300 model, modify its classification head for our custom classes, train it, save the model, and run inference with bounding box visualization on test images.
🔧 Tech Used: PyTorch, Torchvision, OpenCV, Matplotlib, Roboflow
💾 GitHub Code: : https://github.com/NitinCVOrbit/Furni...
#AI #DeepLearning #ComputerVision #ObjectDetection #PyTorch #SSD #MachineLearning #Python #Roboflow #DataScience