{ "CUDNN_BENCHMARK": false, "DATALOADER": { "ASPECT_RATIO_GROUPING": true, "FILTER_EMPTY_ANNOTATIONS": true, "NUM_WORKERS": 4, "REPEAT_SQRT": true, "REPEAT_THRESHOLD": 0.0, "SAMPLER_TRAIN": "TrainingSampler" }, "DATASETS": { "PRECOMPUTED_PROPOSAL_TOPK_TEST": 1000, "PRECOMPUTED_PROPOSAL_TOPK_TRAIN": 2000, "PROPOSAL_FILES_TEST": [], "PROPOSAL_FILES_TRAIN": [], "TEST": [ "keypoints_coco_2017_val" ], "TRAIN": [ "keypoints_coco_2017_train" ] }, "FLOAT32_PRECISION": "", "GLOBAL": { "HACK": 1.0 }, "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.0, 2.0 ] ], "NAME": "DefaultAnchorGenerator", "OFFSET": 0.0, "SIZES": [ [ 32 ], [ 64 ], [ 128 ], [ 256 ], [ 512 ] ] }, "BACKBONE": { "FREEZE_AT": 2, "NAME": "build_resnet_fpn_backbone" }, "DEVICE": "cpu", "FPN": { "FUSE_TYPE": "sum", "IN_FEATURES": [ "res2", "res3", "res4", "res5" ], "NORM": "", "OUT_CHANNELS": 256 }, "KEYPOINT_ON": true, "LOAD_PROPOSALS": false, "MASK_ON": false, "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.0 }, "PIXEL_MEAN": [ 103.53, 116.28, 123.675 ], "PIXEL_STD": [ 1.0, 1.0, 1.0 ], "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.0, 1.0, 1.0, 1.0 ], "FOCAL_LOSS_ALPHA": 0.25, "FOCAL_LOSS_GAMMA": 2.0, "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.0, 10.0, 5.0, 5.0 ], [ 20.0, 20.0, 10.0, 10.0 ], [ 30.0, 30.0, 15.0, 15.0 ] ], "IOUS": [ 0.5, 0.6, 0.7 ] }, "ROI_BOX_HEAD": { "BBOX_REG_LOSS_TYPE": "smooth_l1", "BBOX_REG_LOSS_WEIGHT": 1.0, "BBOX_REG_WEIGHTS": [ 10.0, 10.0, 5.0, 5.0 ], "CLS_AGNOSTIC_BBOX_REG": false, "CONV_DIM": 256, "FC_DIM": 1024, "FED_LOSS_FREQ_WEIGHT_POWER": 0.5, "FED_LOSS_NUM_CLASSES": 50, "NAME": "FastRCNNConvFCHead", "NORM": "", "NUM_CONV": 0, "NUM_FC": 2, "POOLER_RESOLUTION": 7, "POOLER_SAMPLING_RATIO": 0, "POOLER_TYPE": "ROIAlignV2", "SMOOTH_L1_BETA": 0.5, "TRAIN_ON_PRED_BOXES": false, "USE_FED_LOSS": false, "USE_SIGMOID_CE": 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": 13, "POSITIVE_FRACTION": 0.25, "PROPOSAL_APPEND_GT": true, "SCORE_THRESH_TEST": 0.1 }, "ROI_KEYPOINT_HEAD": { "CONV_DIMS": [ 512, 512, 512, 512, 512, 512, 512, 512 ], "LOSS_WEIGHT": 1.0, "MIN_KEYPOINTS_PER_IMAGE": 1, "NAME": "KRCNNConvDeconvUpsampleHead", "NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS": true, "NUM_KEYPOINTS": 10, "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.0, "BBOX_REG_WEIGHTS": [ 1.0, 1.0, 1.0, 1.0 ], "BOUNDARY_THRESH": -1, "CONV_DIMS": [ -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.0, "NMS_THRESH": 0.7, "POSITIVE_FRACTION": 0.5, "POST_NMS_TOPK_TEST": 1000, "POST_NMS_TOPK_TRAIN": 1500, "PRE_NMS_TOPK_TEST": 1000, "PRE_NMS_TOPK_TRAIN": 2000, "SMOOTH_L1_BETA": 0.0 }, "SEM_SEG_HEAD": { "COMMON_STRIDE": 4, "CONVS_DIM": 128, "IGNORE_VALUE": 255, "IN_FEATURES": [ "p2", "p3", "p4", "p5" ], "LOSS_WEIGHT": 1.0, "NAME": "SemSegFPNHead", "NORM": "GN", "NUM_CLASSES": 54 }, "WEIGHTS": "/content/drive/MyDrive/Loi/output_r50_hybrid_4280img_50epochs_bs10_maxsize1024/model_0017975.pth" }, "OUTPUT_DIR": "/content/drive/MyDrive/Loi/output_r50_hybrid_4280img_50epochs_bs10_maxsize1024", "SEED": -1, "SOLVER": { "AMP": { "ENABLED": false }, "BASE_LR": 0.02, "BASE_LR_END": 0.0, "BIAS_LR_FACTOR": 1.0, "CHECKPOINT_PERIOD": 5000, "CLIP_GRADIENTS": { "CLIP_TYPE": "value", "CLIP_VALUE": 1.0, "ENABLED": false, "NORM_TYPE": 2.0 }, "GAMMA": 0.1, "IMS_PER_BATCH": 16, "LR_SCHEDULER_NAME": "WarmupMultiStepLR", "MAX_ITER": 270000, "MOMENTUM": 0.9, "NESTEROV": false, "NUM_DECAYS": 3, "REFERENCE_WORLD_SIZE": 0, "RESCALE_INTERVAL": false, "STEPS": [ 210000, 250000 ], "WARMUP_FACTOR": 0.001, "WARMUP_ITERS": 1000, "WARMUP_METHOD": "linear", "WEIGHT_DECAY": 0.0001, "WEIGHT_DECAY_BIAS": null, "WEIGHT_DECAY_NORM": 0.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": [ 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0 ], "PRECISE_BN": { "ENABLED": false, "NUM_ITER": 200 } }, "VERSION": 2, "VIS_PERIOD": 0, "id2label": { "0": "diamonds", "1": "long_oval", "2": "parallelogram", "3": "rectangle", "4": "circle", "5": "triangle", "6": "object", "7": "hexagon", "8": "trapezoid", "9": "text", "10": "arrow", "11": "line", "12": "double_arrow" } }