aadex commited on
Commit
9e7840d
·
verified ·
1 Parent(s): d229a09

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +71 -0
README.md ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - object-detection
5
+ - vision-transformer
6
+ - coco
7
+ - faster-rcnn
8
+ - positional-embeddings
9
+ - simple-vit
10
+ datasets:
11
+ - coco
12
+ library_name: mmdetection
13
+ ---
14
+
15
+ # Simple ViT - Object Detection on COCO
16
+
17
+ Faster R-CNN with Simple ViT-Tiny backbone (learned positional embeddings)
18
+
19
+ ## Model Details
20
+
21
+ - **Architecture**: Faster R-CNN with ViT-Tiny backbone
22
+ - **Backbone**: Simple ViT (192-dim, 12 layers, 3 heads)
23
+ - **Positional Embedding**: SIMPLE
24
+ - **Training Resolution**: 512x512
25
+ - **Dataset**: COCO 2017
26
+ - **Framework**: MMDetection
27
+
28
+ ## Training Configuration
29
+
30
+ | Parameter | Value |
31
+ |-----------|-------|
32
+ | Image Size | 512x512 |
33
+ | Patch Size | 16x16 |
34
+ | Hidden Dim | 192 |
35
+ | Layers | 12 |
36
+ | Heads | 3 |
37
+ | MLP Dim | 768 |
38
+
39
+ ## Checkpoint Info
40
+
41
+ - **Filename**: `best_coco_bbox_mAP_epoch_12.pth`
42
+ - **Size**: 114.9 MB
43
+ - **Epoch**: 12
44
+
45
+ ## Usage
46
+
47
+ ```python
48
+ from mmdet.apis import init_detector, inference_detector
49
+
50
+ config_file = 'detection/configs/faster_rcnn_simple_vit_tiny_coco.py'
51
+ checkpoint_file = 'best_coco_bbox_mAP_epoch_12.pth'
52
+
53
+ model = init_detector(config_file, checkpoint_file, device='cuda:0')
54
+ result = inference_detector(model, 'test.jpg')
55
+ ```
56
+
57
+ ## Citation
58
+
59
+ If you use this model, please cite:
60
+
61
+ ```bibtex
62
+ @misc{vit_detection_coco,
63
+ title={Vision Transformer Object Detection with Simple ViT},
64
+ year={2026},
65
+ publisher={Hugging Face},
66
+ }
67
+ ```
68
+
69
+ ## License
70
+
71
+ Apache 2.0