Modelpublic

Goutham204/Object_detection

sourceHugging Facemitupdated 1y agoView on Hugging Face

0likes

Model Card

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

text

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