DATA:
  DATA_LOADER: # need
    DATA_DIR: '/fileser/DATA/IMAGE/PROJECT/Nodule/Det/LUNG_CUTTING_SPACING_NODULE_1.0_LINEAR_NoduleNet'
    TRAIN_DB: '/fileser51/DataRadiology/LABEL/PROJECT/Nodule/detect/DB/gold_505_183/raw_spacing'
    VALIDATE_DB: '/fileser/DATA/LABEL/PROJECT/Nodule/detect/FPN/task_1_2_3_4_5_6_7_8_drop/1.0_spacing_lung_region/val_split_by_nidi'
    TEST_DB: '/fileser/DATA/LABEL/PROJECT/Nodule/detect/grouplung_version/202007/gold/gold_183/1.0_spacing_lung_region'
    RAND_CROP_RATIO: 0.0
    NUM_WORKERS: 12
    BALANCED_SAMPLING: True
    POS_TARGET_RANGE: [1,2,3,4,5,6,7,26]
    NEG_TARGET_RANGE: []
  
  DATA_PROCESS: # need
    NUM_NEG: 800
    NUM_HARD: 2
    PAD_VALUE: -1000
    DIAMETERLIM: [12., 100.]
    SPACING: [1.0, 1.0, 1.0]
    FPN_DIAMETER_RANGE: [[20, 99999], [12, 32], [8, 16]]
    AUGTYPE:
      FLIP: True
      ROTATE: True
      SCALE: True
      SWAP: False

MODEL: # need
  WEIGHT: None
  BBOX_REG_WEIGHT: [1., 1., 1., 1., 1., 1.]


  ANCHOR:
#   BASES: [5, 20, 40, 60, 90]
    BASES: [[30, 55, 90], [20, 30, 40], [5, 10, 20]]
    ASPECT_RATIOS: [[1.0, 1.0, 1.0]]
  
  BACKBONE:
    BN_MOMENTUM : 0.1
    FPN: True

  RPN:
    BG_THRESH_HIGH: 0.2
    FG_THRESH_LOW: 0.5
    TRAIN_PRE_NMS_SCORE_THRESH: 0.2
    TEST_PRE_NMS_SCORE_THRESH: 0.05
  
  ROI_BOX_HEAD:
    ROI_BATCH_SIZE: 16
    BG_THRESH_HIGH: 0.02
    FG_THRESH_LOW: 0.7
    FG_FRACTION: 0.5
    TEST_PRE_NMS_SCORE_THRESH: 0

TRAINING:
  NUM_GPUS: 4
  AMP: True
  BATCH_SIZE: 2

  SOLVER:
    OPTIMIZER: 'SGD'
    BASE_LR: 0.005
    MOMENTUM: 0.9
    WRIGHT_DECAY: 1e-4


  SAVER: # need
    SAVER_DIR: './results'

  SHEDULER: # need
    TOTAL_EPOCHS: 200
    SWITCH_ROI_EPOCH: 201
    SWITCH_BALANCED_SAMPLING_EPOCH: 201
    LR_SHEDULE: True
    NUM_NEG_SHEDULE: False
    RAND_CROP_RATIO_SHEDULE: False

TESTING:
  WEIGHT: "/data/huangky/FPN/20201215/checkpoints/060.pth"
  SAVER_DIR: '/data/huangky/FPN/20201215/res'
  USE_RCNN: False

DEPLOY:
  TORCHSCRIPT_SAVE_PATH: "/data/huangky/FPN/20201215/fpn.pt"