{ "data": { "data_loader": { "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, "blacklist": [], "balanced_sampling": true, "pos_target_range": [ 1, 2, 3, 4, 5, 6, 7, 26 ], "neg_target_range": [], "irrelevant_target_range": [], "data_mode": "rib", "mask_dir": "" }, "data_process": { "crop_size": [ 128, 128, 128 ], "spacing": [ 1.0, 1.0, 1.0 ], "bbox_border": 0, "pad_value": -1000, "stride": 4, "num_neg": 800, "num_hard": 2, "bound_size": 8, "diameterlim": [ 12.0, 100.0 ], "scalelim": [ 0.8, 1.2 ], "augtype": { "flip": true, "rotate": true, "scale": true, "swap": false }, "fpn_diameter_range": [ [ 20, 99999 ], [ 12, 32 ], [ 8, 16 ] ], "hu_min": -300.0, "hu_max": 1000.0, "max_objs": 20, "downsampling_ratio": 2, "min_overlap": 0.1 } }, "model": { "meta_architecture": "nodulenet", "weight": "", "bbox_reg_weight": [ 1.0, 1.0, 1.0, 1.0, 1.0, 1.0 ], "anchor": { "bases": [ [ 30, 55, 90 ], [ 20, 30, 40 ], [ 5, 10, 20 ] ], "aspect_ratios": [ [ 1.0, 1.0, 1.0 ] ] }, "backbone": { "conv_body": "resunet", "bn_momentum": 0.1, "fpn": true }, "rpn": { "use_fpn": false, "bg_thresh_high": 0.2, "fg_thresh_low": 0.5, "nms_iou_thresh": 0.1, "train_pre_nms_score_thresh": 0.2, "test_pre_nms_score_thresh": 0.05, "deploy_pre_nms_top_k": 10000, "balanced_cls_loss": true }, "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": 2, "solver": { "optimizer": "sgd", "base_lr": 0.005, "momentum": 0.9, "wright_decay": 0.0001 }, "saver": { "saver_dir": "./results", "saver_frequency": 1 }, "sheduler": { "total_epochs": 200, "warmup": 5, "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", "torchscript_compare_res_dir": "" } }