Goutham204 commited on
Commit
c20cd65
·
verified ·
1 Parent(s): b29dd97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -3
README.md CHANGED
@@ -1,3 +1,43 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - object-detection
6
+ - pytorch
7
+ - fasterrcnn
8
+ - computer-vision
9
+ - pascalvoc
10
+ model-index:
11
+ - name: Faster R-CNN Object Detection (Pascal VOC)
12
+ results:
13
+ - task:
14
+ type: object-detection
15
+ name: Object Detection
16
+ dataset:
17
+ name: Pascal VOC 2012
18
+ type: pascal_voc
19
+ metrics:
20
+ - name: mAP
21
+ type: mean_average_precision
22
+ value: 0.72
23
+ ---
24
+
25
+ # Faster R-CNN Object Detection (Pascal VOC - PyTorch)
26
+
27
+ 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.
28
+
29
+ ## Model Info
30
+
31
+ - **Architecture**: Faster R-CNN with ResNet-50 FPN
32
+ - **Framework**: PyTorch
33
+ - **Dataset**: Pascal VOC 2012
34
+ - **Number of Classes**: 20 (excluding background)
35
+ - **File Format**: `fasterrcnn_model.pth`
36
+
37
+ ## Object Classes
38
+
39
+ ```text
40
+ ['aeroplane', 'bicycle', 'bird', 'boat', 'bottle',
41
+ 'bus', 'car', 'cat', 'chair', 'cow',
42
+ 'diningtable', 'dog', 'horse', 'motorbike', 'person',
43
+ 'pottedplant', 'sheep', 'sofa', 'train', 'tvmonitor']