fuliucansheng
commited on
Commit
•
45ca6b2
1
Parent(s):
67bbdd3
add faster rcnn & vlp model
Browse files- .gitattributes +3 -0
- faster-rcnn/pascal-voc-detection/faster_rcnn_r50_c4_config.yaml +13 -0
- faster-rcnn/pascal-voc-detection/faster_rcnn_r50_c4_model.bin +3 -0
- vlp/config.json +18 -0
- vlp/faster_rcnn_x101_64x4d_fpn_2x_config.yaml +42 -0
- vlp/faster_rcnn_x101_64x4d_fpn_2x_model.bin +3 -0
- vlp/pytorch_model.bin +3 -0
.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
|