fuliucansheng commited on
Commit
45ca6b2
1 Parent(s): 67bbdd3

add faster rcnn & vlp model

Browse files
.gitattributes CHANGED
@@ -29,3 +29,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
29
  ./yolo/v5/yolov5m.bin filter=lfs diff=lfs merge=lfs -text
30
  ./yolo/v5/yolov5s.bin filter=lfs diff=lfs merge=lfs -text
31
  ./yolo/v5/yolov5x.bin filter=lfs diff=lfs merge=lfs -text
 
 
 
 
29
  ./yolo/v5/yolov5m.bin filter=lfs diff=lfs merge=lfs -text
30
  ./yolo/v5/yolov5s.bin filter=lfs diff=lfs merge=lfs -text
31
  ./yolo/v5/yolov5x.bin filter=lfs diff=lfs merge=lfs -text
32
+ ./vlp/pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
33
+ ./vlp/faster_rcnn_x101_64x4d_fpn_2x_model.bin filter=lfs diff=lfs merge=lfs -text
34
+ ./faster-rcnn/pascal-voc-detection/faster_rcnn_r50_c4_model.bin filter=lfs diff=lfs merge=lfs -text
faster-rcnn/pascal-voc-detection/faster_rcnn_r50_c4_config.yaml ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MODEL:
2
+ META_ARCHITECTURE: "GeneralizedRCNN"
3
+ RPN:
4
+ PRE_NMS_TOPK_TEST: 6000
5
+ POST_NMS_TOPK_TEST: 1000
6
+ ROI_HEADS:
7
+ NAME: "Res5ROIHeads"
8
+ NUM_CLASSES: 20
9
+ MASK_ON: False
10
+ RESNETS:
11
+ DEPTH: 50
12
+ VERSION: 2
13
+
faster-rcnn/pascal-voc-detection/faster_rcnn_r50_c4_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6abec3dd4c28fc7a049add215e7877ba9bcf228f4f9e1150580b0a6abbe62790
3
+ size 133221355
vlp/config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attention_probs_dropout_prob": 0.1,
3
+ "directionality": "bidi",
4
+ "hidden_act": "gelu",
5
+ "hidden_dropout_prob": 0.1,
6
+ "hidden_size": 768,
7
+ "initializer_range": 0.02,
8
+ "intermediate_size": 3072,
9
+ "max_position_embeddings": 512,
10
+ "num_attention_heads": 12,
11
+ "num_hidden_layers": 12,
12
+ "type_vocab_size": 6,
13
+ "layer_norm_epsilon": 0.00001,
14
+ "vocab_size": 28996,
15
+ "label_smoothing": 0.0,
16
+ "source_type_id": 4,
17
+ "target_type_id": 5
18
+ }
vlp/faster_rcnn_x101_64x4d_fpn_2x_config.yaml ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MODEL:
2
+ META_ARCHITECTURE: "GeneralizedRCNN"
3
+ MASK_ON: False
4
+ BACKBONE:
5
+ NAME: "build_resnet_fpn_backbone"
6
+ RESNETS:
7
+ OUT_FEATURES: ["res2", "res3", "res4", "res5"]
8
+ STRIDE_IN_1X1: False # this is a C2 model
9
+ NUM_GROUPS: 64
10
+ WIDTH_PER_GROUP: 4
11
+ DEPTH: 101
12
+ FPN:
13
+ IN_FEATURES: ["res2", "res3", "res4", "res5"]
14
+ OUT_CHANNELS: 512
15
+ ANCHOR_GENERATOR:
16
+ SIZES: [[32], [64], [128], [256], [512]] # One size for each in feature map
17
+ ASPECT_RATIOS: [[0.5, 1.0, 2.0]] # Three aspect ratios (same for all in feature maps)
18
+ RPN:
19
+ IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
20
+ PRE_NMS_TOPK_TRAIN: 2000 # Per FPN level
21
+ PRE_NMS_TOPK_TEST: 1000 # Per FPN level
22
+ # Detectron1 uses 2000 proposals per-batch,
23
+ # (See "modeling/rpn/rpn_outputs.py" for details of this legacy issue)
24
+ # which is approximately 1000 proposals per-image since the default batch size for FPN is 2.
25
+ POST_NMS_TOPK_TRAIN: 1000
26
+ POST_NMS_TOPK_TEST: 1000
27
+ NMS_THRESH: 0.7
28
+ ROI_HEADS:
29
+ NAME: "StandardROIHeads"
30
+ IN_FEATURES: ["p2", "p3", "p4", "p5"]
31
+ NUM_CLASSES: 1601
32
+ NMS_THRESH_TEST: 0.3
33
+ ROI_BOX_HEAD:
34
+ NAME: "FastRCNNConvFCHead"
35
+ NUM_CONV: 0
36
+ NUM_FC: 2
37
+ FC_DIM: 2048
38
+ POOLER_RESOLUTION: 7
39
+ POOLER_SAMPLING_RATIO: 2
40
+ POOLER_TYPE: "ROIAlign"
41
+ VERSION: 2
42
+
vlp/faster_rcnn_x101_64x4d_fpn_2x_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ad57a9ca8cc157cc7d7bbe052602277ce4feed757b90efcaa050cdfee76d38d
3
+ size 668770123
vlp/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e93d44ee113e6215d955646b4ed17539aad5deb5a898259d4120d23224520dd5
3
+ size 463813740