Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
hackassin
GitHub Repository: hackassin/learnopencv
 Path: blob/master/FaceMaskOverlay/experiments/300w/face_alignment_300w_hrnet_w18.yaml
3168 views
1
GPUS: (0, )

2
WORKERS: 4

3
PRINT_FREQ: 50

4


5
DATASET:

6
  DATASET: 300W

7
  ROOT: './data/300w/images'

8
  TRAINSET: './data/300w/face_landmarks_300w_train.csv'

9
  TESTSET: './data/300w/face_landmarks_300w_valid.csv'

10
  FLIP: true

11
  SCALE_FACTOR: 0.25

12
  ROT_FACTOR: 30

13
MODEL:

14
  NAME: 'hrnet'

15
  NUM_JOINTS: 68

16
  INIT_WEIGHTS: true

17
  PRETRAINED: 'hrnetv2_pretrained/hrnetv2_w18_imagenet_pretrained.pth'

18
  SIGMA: 1.0 

19
  IMAGE_SIZE:

20
    - 256

21
    - 256

22
  HEATMAP_SIZE:

23
    - 64

24
    - 64

25
  MEAN:

26
    - 0.485

27
    - 0.456

28
    - 0.406

29
  STD:

30
    - 0.229

31
    - 0.224

32
    - 0.225

33
  EXTRA:

34
    FINAL_CONV_KERNEL: 1

35
    STAGE2:

36
      NUM_MODULES: 1

37
      NUM_BRANCHES: 2

38
      BLOCK: BASIC

39
      NUM_BLOCKS:

40
        - 4

41
        - 4

42
      NUM_CHANNELS:

43
        - 18

44
        - 36

45
      FUSE_METHOD: SUM

46
    STAGE3:

47
      NUM_MODULES: 4

48
      NUM_BRANCHES: 3

49
      BLOCK: BASIC

50
      NUM_BLOCKS:

51
        - 4

52
        - 4

53
        - 4

54
      NUM_CHANNELS:

55
        - 18

56
        - 36

57
        - 72

58
      FUSE_METHOD: SUM

59
    STAGE4:

60
      NUM_MODULES: 3

61
      NUM_BRANCHES: 4

62
      BLOCK: BASIC

63
      NUM_BLOCKS:

64
        - 4

65
        - 4

66
        - 4

67
        - 4

68
      NUM_CHANNELS:

69
        - 18

70
        - 36

71
        - 72

72
        - 144

73
      FUSE_METHOD: SUM

74
TRAIN:

75
  BATCH_SIZE_PER_GPU: 16

76
  SHUFFLE: true

77
  BEGIN_EPOCH: 0

78
  END_EPOCH: 60

79
  RESUME: true

80
  OPTIMIZER: 'adam'

81
  LR: 0.0001

82
  WD: 0.0

83
  LR_STEP: [30, 50]

84
  MOMENTUM: 0.0

85
  NESTEROV: false

86
TEST:

87
  BATCH_SIZE_PER_GPU: 8

88


89