{ "CUDNN_BENCHMARK": false, "DATALOADER": { "ASPECT_RATIO_GROUPING": true, "FILTER_EMPTY_ANNOTATIONS": true, "NUM_WORKERS": 4, "REPEAT_THRESHOLD": 0, "SAMPLER_TRAIN": "TrainingSampler" }, "DATASETS": { "PRECOMPUTED_PROPOSAL_TOPK_TEST": 1000, "PRECOMPUTED_PROPOSAL_TOPK_TRAIN": 2000, "PROPOSAL_FILES_TEST": [], "PROPOSAL_FILES_TRAIN": [], "TEST": [ "modele-val" ], "TRAIN": [ "modele-train" ] }, "GLOBAL": { "HACK": 1 }, "INPUT": { "CROP": { "ENABLED": false, "SIZE": [ 0.9, 0.9 ], "TYPE": "relative_range" }, "FORMAT": "BGR", "MASK_FORMAT": "polygon", "MAX_SIZE_TEST": 1333, "MAX_SIZE_TRAIN": 1333, "MIN_SIZE_TEST": 800, "MIN_SIZE_TRAIN": [ 640, 672, 704, 736, 768, 800 ], "MIN_SIZE_TRAIN_SAMPLING": "choice", "RANDOM_FLIP": "horizontal" }, "MODEL": { "ANCHOR_GENERATOR": { "ANGLES": [ [ -90, 0, 90 ] ], "ASPECT_RATIOS": [ [ 0.5, 1, 2 ] ], "NAME": "DefaultAnchorGenerator", "OFFSET": 0, "SIZES": [ [ 32 ], [ 64 ], [ 128 ], [ 256 ], [ 512 ] ] }, "BACKBONE": { "FREEZE_AT": 2, "NAME": "build_resnet_fpn_backbone" }, "DEVICE": "cuda", "FPN": { "FUSE_TYPE": "sum", "IN_FEATURES": [ "res2", "res3", "res4", "res5" ], "NORM": "", "OUT_CHANNELS": 256 }, "KEYPOINT_ON": false, "LOAD_PROPOSALS": false, "MASK_ON": true, "META_ARCHITECTURE": "GeneralizedRCNN", "PANOPTIC_FPN": { "COMBINE": { "ENABLED": true, "INSTANCES_CONFIDENCE_THRESH": 0.5, "OVERLAP_THRESH": 0.5, "STUFF_AREA_LIMIT": 4096 }, "INSTANCE_LOSS_WEIGHT": 1 }, "PIXEL_MEAN": [ 103.53, 116.28, 123.675 ], "PIXEL_STD": [ 1, 1, 1 ], "PROPOSAL_GENERATOR": { "MIN_SIZE": 0, "NAME": "RPN" }, "RESNETS": { "DEFORM_MODULATED": false, "DEFORM_NUM_GROUPS": 1, "DEFORM_ON_PER_STAGE": [ false, false, false, false ], "DEPTH": 50, "NORM": "FrozenBN", "NUM_GROUPS": 1, "OUT_FEATURES": [ "res2", "res3", "res4", "res5" ], "RES2_OUT_CHANNELS": 256, "RES5_DILATION": 1, "STEM_OUT_CHANNELS": 64, "STRIDE_IN_1X1": true, "WIDTH_PER_GROUP": 64 }, "RETINANET": { "BBOX_REG_LOSS_TYPE": "smooth_l1", "BBOX_REG_WEIGHTS": [ 1, 1, 1, 1 ], "FOCAL_LOSS_ALPHA": 0.25, "FOCAL_LOSS_GAMMA": 2, "IN_FEATURES": [ "p3", "p4", "p5", "p6", "p7" ], "IOU_LABELS": [ 0, -1, 1 ], "IOU_THRESHOLDS": [ 0.4, 0.5 ], "NMS_THRESH_TEST": 0.5, "NORM": "", "NUM_CLASSES": 80, "NUM_CONVS": 4, "PRIOR_PROB": 0.01, "SCORE_THRESH_TEST": 0.05, "SMOOTH_L1_LOSS_BETA": 0.1, "TOPK_CANDIDATES_TEST": 1000 }, "ROI_BOX_CASCADE_HEAD": { "BBOX_REG_WEIGHTS": [ [ 10, 10, 5, 5 ], [ 20, 20, 10, 10 ], [ 30, 30, 15, 15 ] ], "IOUS": [ 0.5, 0.6, 0.7 ] }, "ROI_BOX_HEAD": { "BBOX_REG_LOSS_TYPE": "smooth_l1", "BBOX_REG_LOSS_WEIGHT": 1, "BBOX_REG_WEIGHTS": [ 10, 10, 5, 5 ], "CLS_AGNOSTIC_BBOX_REG": false, "CONV_DIM": 256, "FC_DIM": 1024, "NAME": "FastRCNNConvFCHead", "NORM": "", "NUM_CONV": 0, "NUM_FC": 2, "POOLER_RESOLUTION": 7, "POOLER_SAMPLING_RATIO": 0, "POOLER_TYPE": "ROIAlignV2", "SMOOTH_L1_BETA": 0, "TRAIN_ON_PRED_BOXES": false }, "ROI_HEADS": { "BATCH_SIZE_PER_IMAGE": 512, "IN_FEATURES": [ "p2", "p3", "p4", "p5" ], "IOU_LABELS": [ 0, 1 ], "IOU_THRESHOLDS": [ 0.5 ], "NAME": "StandardROIHeads", "NMS_THRESH_TEST": 0.5, "NUM_CLASSES": 2, "POSITIVE_FRACTION": 0.25, "PROPOSAL_APPEND_GT": true, "SCORE_THRESH_TEST": 0.05 }, "ROI_KEYPOINT_HEAD": { "CONV_DIMS": [ 512, 512, 512, 512, 512, 512, 512, 512 ], "LOSS_WEIGHT": 1, "MIN_KEYPOINTS_PER_IMAGE": 1, "NAME": "KRCNNConvDeconvUpsampleHead", "NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS": true, "NUM_KEYPOINTS": 17, "POOLER_RESOLUTION": 14, "POOLER_SAMPLING_RATIO": 0, "POOLER_TYPE": "ROIAlignV2" }, "ROI_MASK_HEAD": { "CLS_AGNOSTIC_MASK": false, "CONV_DIM": 256, "NAME": "MaskRCNNConvUpsampleHead", "NORM": "", "NUM_CONV": 4, "POOLER_RESOLUTION": 14, "POOLER_SAMPLING_RATIO": 0, "POOLER_TYPE": "ROIAlignV2" }, "RPN": { "BATCH_SIZE_PER_IMAGE": 256, "BBOX_REG_LOSS_TYPE": "smooth_l1", "BBOX_REG_LOSS_WEIGHT": 1, "BBOX_REG_WEIGHTS": [ 1, 1, 1, 1 ], "BOUNDARY_THRESH": -1, "HEAD_NAME": "StandardRPNHead", "IN_FEATURES": [ "p2", "p3", "p4", "p5", "p6" ], "IOU_LABELS": [ 0, -1, 1 ], "IOU_THRESHOLDS": [ 0.3, 0.7 ], "LOSS_WEIGHT": 1, "NMS_THRESH": 0.7, "POSITIVE_FRACTION": 0.5, "POST_NMS_TOPK_TEST": 1000, "POST_NMS_TOPK_TRAIN": 1000, "PRE_NMS_TOPK_TEST": 1000, "PRE_NMS_TOPK_TRAIN": 2000, "SMOOTH_L1_BETA": 0 }, "SEM_SEG_HEAD": { "COMMON_STRIDE": 4, "CONVS_DIM": 128, "IGNORE_VALUE": 255, "IN_FEATURES": [ "p2", "p3", "p4", "p5" ], "LOSS_WEIGHT": 1, "NAME": "SemSegFPNHead", "NORM": "GN", "NUM_CLASSES": 54 }, "WEIGHTS": "/content/drive/MyDrive/layoutparser/modele/modele3_NP?/model_final.pth" }, "OUTPUT_DIR": "/content/drive/MyDrive/layoutparser/modele", "SEED": -1, "SOLVER": { "AMP": { "ENABLED": false }, "BASE_LR": 0.00025, "BIAS_LR_FACTOR": 1, "CHECKPOINT_PERIOD": 50, "CLIP_GRADIENTS": { "CLIP_TYPE": "value", "CLIP_VALUE": 1, "ENABLED": false, "NORM_TYPE": 2 }, "GAMMA": 0.1, "IMS_PER_BATCH": 2, "LR_SCHEDULER_NAME": "WarmupMultiStepLR", "MAX_ITER": 300, "MOMENTUM": 0.9, "NESTEROV": false, "REFERENCE_WORLD_SIZE": 0, "STEPS": [ 210000, 250000 ], "WARMUP_FACTOR": 0.001, "WARMUP_ITERS": 1000, "WARMUP_METHOD": "linear", "WEIGHT_DECAY": 0.0001, "WEIGHT_DECAY_BIAS": 0.0001, "WEIGHT_DECAY_NORM": 0 }, "TEST": { "AUG": { "ENABLED": false, "FLIP": true, "MAX_SIZE": 4000, "MIN_SIZES": [ 400, 500, 600, 700, 800, 900, 1000, 1100, 1200 ] }, "DETECTIONS_PER_IMAGE": 100, "EVAL_PERIOD": 0, "EXPECTED_RESULTS": [], "KEYPOINT_OKS_SIGMAS": [], "PRECISE_BN": { "ENABLED": false, "NUM_ITER": 200 } }, "VERSION": 2, "VIS_PERIOD": 0 }