{ "_name_or_path": "microsoft/conditional-detr-resnet-50", "activation_dropout": 0.0, "activation_function": "relu", "architectures": [ "ConditionalDetrForObjectDetection" ], "attention_dropout": 0.0, "auxiliary_loss": false, "backbone": "resnet50", "backbone_config": null, "backbone_kwargs": { "in_chans": 3, "out_indices": [ 1, 2, 3, 4 ] }, "bbox_cost": 5, "bbox_loss_coefficient": 5, "class_cost": 2, "cls_loss_coefficient": 2, "d_model": 256, "decoder_attention_heads": 8, "decoder_ffn_dim": 2048, "decoder_layerdrop": 0.0, "decoder_layers": 6, "dice_loss_coefficient": 1, "dilation": false, "dropout": 0.1, "encoder_attention_heads": 8, "encoder_ffn_dim": 2048, "encoder_layerdrop": 0.0, "encoder_layers": 6, "focal_alpha": 0.25, "giou_cost": 2, "giou_loss_coefficient": 2, "id2label": { "0": "Spinal Canal Stenosis L1/L2", "1": "Spinal Canal Stenosis L2/L3", "2": "Spinal Canal Stenosis L3/L4", "3": "Spinal Canal Stenosis L4/L5", "4": "Spinal Canal Stenosis L5/S1", "5": "Right Neural Foraminal Narrowing L4/L5", "6": "Right Neural Foraminal Narrowing L5/S1", "7": "Right Neural Foraminal Narrowing L3/L4", "8": "Right Neural Foraminal Narrowing L1/L2", "9": "Right Neural Foraminal Narrowing L2/L3", "10": "Left Neural Foraminal Narrowing L1/L2", "11": "Left Neural Foraminal Narrowing L4/L5", "12": "Left Neural Foraminal Narrowing L5/S1", "13": "Left Neural Foraminal Narrowing L2/L3", "14": "Left Neural Foraminal Narrowing L3/L4", "15": "Left Subarticular Stenosis L1/L2", "16": "Right Subarticular Stenosis L1/L2", "17": "Left Subarticular Stenosis L2/L3", "18": "Right Subarticular Stenosis L2/L3", "19": "Left Subarticular Stenosis L3/L4", "20": "Right Subarticular Stenosis L3/L4", "21": "Left Subarticular Stenosis L4/L5", "22": "Right Subarticular Stenosis L4/L5", "23": "Left Subarticular Stenosis L5/S1", "24": "Right Subarticular Stenosis L5/S1" }, "init_std": 0.02, "init_xavier_std": 1.0, "is_encoder_decoder": true, "label2id": { "Left Neural Foraminal Narrowing L1/L2": 10, "Left Neural Foraminal Narrowing L2/L3": 13, "Left Neural Foraminal Narrowing L3/L4": 14, "Left Neural Foraminal Narrowing L4/L5": 11, "Left Neural Foraminal Narrowing L5/S1": 12, "Left Subarticular Stenosis L1/L2": 15, "Left Subarticular Stenosis L2/L3": 17, "Left Subarticular Stenosis L3/L4": 19, "Left Subarticular Stenosis L4/L5": 21, "Left Subarticular Stenosis L5/S1": 23, "Right Neural Foraminal Narrowing L1/L2": 8, "Right Neural Foraminal Narrowing L2/L3": 9, "Right Neural Foraminal Narrowing L3/L4": 7, "Right Neural Foraminal Narrowing L4/L5": 5, "Right Neural Foraminal Narrowing L5/S1": 6, "Right Subarticular Stenosis L1/L2": 16, "Right Subarticular Stenosis L2/L3": 18, "Right Subarticular Stenosis L3/L4": 20, "Right Subarticular Stenosis L4/L5": 22, "Right Subarticular Stenosis L5/S1": 24, "Spinal Canal Stenosis L1/L2": 0, "Spinal Canal Stenosis L2/L3": 1, "Spinal Canal Stenosis L3/L4": 2, "Spinal Canal Stenosis L4/L5": 3, "Spinal Canal Stenosis L5/S1": 4 }, "mask_loss_coefficient": 1, "max_position_embeddings": 1024, "model_type": "conditional_detr", "num_channels": 3, "num_hidden_layers": 6, "num_queries": 300, "position_embedding_type": "sine", "scale_embedding": false, "torch_dtype": "float32", "transformers_version": "4.41.2", "use_pretrained_backbone": true, "use_timm_backbone": true }