File size: 1,213 Bytes
c20cd65 7922b5d c20cd65 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
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
```text
['aeroplane', 'bicycle', 'bird', 'boat', 'bottle',
'bus', 'car', 'cat', 'chair', 'cow',
'diningtable', 'dog', 'horse', 'motorbike', 'person',
'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor']
|