artmiss/trashify-object-detection-model

1{2  "activation_dropout": 0.0,3  "activation_function": "silu",4  "anchor_image_size": null,5  "architectures": [6    "RTDetrV2ForObjectDetection"7  ],8  "attention_dropout": 0.0,9  "auxiliary_loss": true,10  "backbone": null,11  "backbone_config": {12    "depths": [13      3,14      4,15      6,16      317    ],18    "downsample_in_bottleneck": false,19    "downsample_in_first_stage": false,20    "dtype": "float32",21    "embedding_size": 64,22    "hidden_act": "relu",23    "hidden_sizes": [24      256,25      512,26      1024,27      204828    ],29    "layer_type": "bottleneck",30    "model_type": "rt_detr_resnet",31    "num_channels": 3,32    "out_features": [33      "stage2",34      "stage3",35      "stage4"36    ],37    "out_indices": [38      2,39      3,40      441    ],42    "stage_names": [43      "stem",44      "stage1",45      "stage2",46      "stage3",47      "stage4"48    ]49  },50  "backbone_kwargs": null,51  "batch_norm_eps": 1e-05,52  "box_noise_scale": 1.0,53  "d_model": 256,54  "decoder_activation_function": "relu",55  "decoder_attention_heads": 8,56  "decoder_ffn_dim": 1024,57  "decoder_in_channels": [58    256,59    256,60    25661  ],62  "decoder_layers": 6,63  "decoder_method": "default",64  "decoder_n_levels": 3,65  "decoder_n_points": 4,66  "decoder_offset_scale": 0.5,67  "disable_custom_kernels": true,68  "dropout": 0.0,69  "dtype": "float32",70  "encode_proj_layers": [71    272  ],73  "encoder_activation_function": "gelu",74  "encoder_attention_heads": 8,75  "encoder_ffn_dim": 1024,76  "encoder_hidden_dim": 256,77  "encoder_in_channels": [78    512,79    1024,80    204881  ],82  "encoder_layers": 1,83  "eos_coefficient": 0.0001,84  "eval_size": null,85  "feat_strides": [86    8,87    16,88    3289  ],90  "focal_loss_alpha": 0.75,91  "focal_loss_gamma": 2.0,92  "freeze_backbone_batch_norms": true,93  "hidden_expansion": 1.0,94  "id2label": {95    "0": "bin",96    "1": "hand",97    "2": "not_bin",98    "3": "not_hand",99    "4": "not_trash",100    "5": "trash",101    "6": "trash_arm"102  },103  "initializer_bias_prior_prob": null,104  "initializer_range": 0.01,105  "is_encoder_decoder": true,106  "label2id": {107    "bin": 0,108    "hand": 1,109    "not_bin": 2,110    "not_hand": 3,111    "not_trash": 4,112    "trash": 5,113    "trash_arm": 6114  },115  "label_noise_ratio": 0.5,116  "layer_norm_eps": 1e-05,117  "learn_initial_query": false,118  "matcher_alpha": 0.25,119  "matcher_bbox_cost": 5.0,120  "matcher_class_cost": 2.0,121  "matcher_gamma": 2.0,122  "matcher_giou_cost": 2.0,123  "model_type": "rt_detr_v2",124  "normalize_before": false,125  "num_denoising": 100,126  "num_feature_levels": 3,127  "num_queries": 300,128  "positional_encoding_temperature": 10000,129  "tie_word_embeddings": true,130  "transformers_version": "5.0.0",131  "use_cache": false,132  "use_focal_loss": true,133  "use_pretrained_backbone": false,134  "use_timm_backbone": false,135  "weight_loss_bbox": 5.0,136  "weight_loss_giou": 2.0,137  "weight_loss_vfl": 1.0,138  "with_box_refine": true139}140