Spaces:
No application file
No application file
Delete Base-DensePose-RCNN-FPN.yaml
Browse files- Base-DensePose-RCNN-FPN.yaml +0 -60
Base-DensePose-RCNN-FPN.yaml
DELETED
@@ -1,60 +0,0 @@
|
|
1 |
-
VERSION: 2
|
2 |
-
MODEL:
|
3 |
-
META_ARCHITECTURE: "GeneralizedRCNN"
|
4 |
-
BACKBONE:
|
5 |
-
NAME: "build_resnet_fpn_backbone"
|
6 |
-
RESNETS:
|
7 |
-
OUT_FEATURES: ["res2", "res3", "res4", "res5"]
|
8 |
-
FPN:
|
9 |
-
IN_FEATURES: ["res2", "res3", "res4", "res5"]
|
10 |
-
ANCHOR_GENERATOR:
|
11 |
-
SIZES: [[32], [64], [128], [256], [512]] # One size for each in feature map
|
12 |
-
ASPECT_RATIOS: [[0.5, 1.0, 2.0]] # Three aspect ratios (same for all in feature maps)
|
13 |
-
RPN:
|
14 |
-
IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
|
15 |
-
PRE_NMS_TOPK_TRAIN: 2000 # Per FPN level
|
16 |
-
PRE_NMS_TOPK_TEST: 1000 # Per FPN level
|
17 |
-
# Detectron1 uses 2000 proposals per-batch,
|
18 |
-
# (See "modeling/rpn/rpn_outputs.py" for details of this legacy issue)
|
19 |
-
# which is approximately 1000 proposals per-image since the default batch size for FPN is 2.
|
20 |
-
POST_NMS_TOPK_TRAIN: 1000
|
21 |
-
POST_NMS_TOPK_TEST: 1000
|
22 |
-
|
23 |
-
DENSEPOSE_ON: True
|
24 |
-
ROI_HEADS:
|
25 |
-
NAME: "DensePoseROIHeads"
|
26 |
-
IN_FEATURES: ["p2", "p3", "p4", "p5"]
|
27 |
-
NUM_CLASSES: 1
|
28 |
-
ROI_BOX_HEAD:
|
29 |
-
NAME: "FastRCNNConvFCHead"
|
30 |
-
NUM_FC: 2
|
31 |
-
POOLER_RESOLUTION: 7
|
32 |
-
POOLER_SAMPLING_RATIO: 2
|
33 |
-
POOLER_TYPE: "ROIAlign"
|
34 |
-
ROI_DENSEPOSE_HEAD:
|
35 |
-
NAME: "DensePoseV1ConvXHead"
|
36 |
-
POOLER_TYPE: "ROIAlign"
|
37 |
-
NUM_COARSE_SEGM_CHANNELS: 2
|
38 |
-
PREDICTOR_NAME: "DensePoseEmbeddingPredictor"
|
39 |
-
LOSS_NAME: "DensePoseCseLoss"
|
40 |
-
CSE:
|
41 |
-
# embedding loss, possible values:
|
42 |
-
# - "EmbeddingLoss"
|
43 |
-
# - "SoftEmbeddingLoss"
|
44 |
-
EMBED_LOSS_NAME: "EmbeddingLoss"
|
45 |
-
SOLVER:
|
46 |
-
IMS_PER_BATCH: 16
|
47 |
-
BASE_LR: 0.01
|
48 |
-
STEPS: (60000, 80000)
|
49 |
-
MAX_ITER: 90000
|
50 |
-
WARMUP_FACTOR: 0.1
|
51 |
-
CLIP_GRADIENTS:
|
52 |
-
CLIP_TYPE: norm
|
53 |
-
CLIP_VALUE: 1.0
|
54 |
-
ENABLED: true
|
55 |
-
NORM_TYPE: 2.0
|
56 |
-
INPUT:
|
57 |
-
MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
|
58 |
-
DENSEPOSE_EVALUATION:
|
59 |
-
TYPE: cse
|
60 |
-
STORAGE: file
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|