DATA:
  DATA_LOADER:
    DATA_DIR: '/home/CT_RIB/mask_bbox_cutted_image/res07'
    TRAIN_DB: '/fileser/CT_RIB/model/0820/rib_fracture_dataset_0820_rib_only/train_wo_mask_res07_bbox_cutting'
    VALIDATE_DB: '/fileser/CT_RIB/model/0820/rib_fracture_dataset_0820_rib_only/validate_wo_mask_res07_bbox_cutting'
    TEST_DB: '/fileser/CT_RIB/model/0820/rib_fracture_dataset_0820_rib_only/test_wo_mask_res07_bbox_cutting'
    RAND_CROP_RATIO: 0.0
    NUM_WORKERS: 12
    BLACKLIST: []
    BALANCED_SAMPLING: False
    POS_TARGET_RANGE: [1,2,3,4,5,6]
    NEG_TARGET_RANGE: []
  
  DATA_PROCESS:
    CROP_SIZE: [128,128,128]
    SPACING: [0.7, 0.7, 0.7]
    BBOX_BORDER: 0
    PAD_VALUE: -300
    STRIDE: 4
    NUM_NEG: 800
    NUM_HARD: 2
    BOUND_SIZE: 8
    DIAMETERLIM: [5., 100.]
    SCALELIM: [0.8, 1.2]
    AUGTYPE:
      FLIP: True
      ROTATE: True
      SCALE: True
      SWAP: False

MODEL:
  META_ARCHITECTURE: 'NoduleNet'
  WEIGHT: None
  BBOX_REG_WEIGHT: [1., 1., 1., 0.5, 0.5, 0.5]


  ANCHOR:
    BASES: [10, 15, 20, 30, 40]
    ASPECT_RATIOS: [[1.0, 1.0, 1.0]]
  
  BACKBONE:
    CONV_BODY: "ResUNet"
    BN_MOMENTUM : 0.1

  RPN:
    USE_FPN: False
    BG_THRESH_HIGH: 0.02
    FG_THRESH_LOW: 0.7
    NMS_IOU_THRESH: 0.1
    TRAIN_PRE_NMS_SCORE_THRESH: 0.2
    TEST_PRE_NMS_SCORE_THRESH: 0.2
    DEPLOY_PRE_NMS_TOP_K: 10000
  
  ROI_BOX_HEAD:
    ROI_BATCH_SIZE: 16
    NUM_CLASS: 2
    POOLER_RESOLUTION : 7
    SAMPLING_RATIO: 2
    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
  FP16_ALLREDUCE: False
  BATCH_SIZE: 10

  SOLVER:
    OPTIMIZER: 'SGD'
    BASE_LR: 0.01
    MOMENTUM: 0.9
    WRIGHT_DECAY: 1e-4
  
  SAVER:
    SAVER_DIR: 'tmp'
    SAVER_FREQUENCY: 1

  
  SHEDULER:
    TOTAL_EPOCHS: 300
    WARMUP: 5
    SWITCH_ROI_EPOCH: 301
    SWITCH_BALANCED_SAMPLING_EPOCH: 301
    LR_SHEDULE: True
    NUM_NEG_SHEDULE: True
    RAND_CROP_RATIO_SHEDULE: True

TESTING:
  WEIGHT: '/fileser/gupc/RibFracture/NoduleNet/rib_10/proxima_data_wo_ribmask/model/234.pth'
  SAVER_DIR: 'res'
  USE_RCNN: False