metadata

language: en
license: mit
tags:
  - object-detection
  - pytorch
  - fasterrcnn
  - computer-vision
  - pascalvoc
model-index:
  - name: Faster R-CNN Object Detection (Pascal VOC)
    results:
      - task:
          type: object-detection
          name: Object Detection
        dataset:
          name: Pascal VOC 2012
          type: pascal_voc
        metrics:
          - name: mAP
            type: mean_average_precision
            value: 0.72

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']