ML-Motivators commited on
Commit
61988b0
·
verified ·
1 Parent(s): a36d755

Delete Base-DensePose-RCNN-FPN.yaml

Browse files
Files changed (1) hide show
  1. 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