Goutham204/Object_detection

Faster R-CNN Object Detection (Pascal VOC - PyTorch)

This repository contains a custom-trained Faster R-CNN object detection model using PyTorch. The model is trained on the Pascal VOC 2012 dataset and detects 20 common object classes in images by drawing bounding boxes with confidence scores.

Model Info

• Architecture: Faster R-CNN with ResNet-50 FPN
• Framework: PyTorch
• Dataset: Pascal VOC 2012
• Number of Classes: 20 (excluding background)
• File Format: objectdetection_model.pth

Object Classes

['aeroplane', 'bicycle', 'bird', 'boat', 'bottle',
 'bus', 'car', 'cat', 'chair', 'cow',
 'diningtable', 'dog', 'horse', 'motorbike', 'person',
 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor']