CoolFace
Modelpublic

cdameron1024/trashify-object-detector

sourceHugging Faceupdated 12d agoView on Hugging Face
0likes31downloads
config.json136 linesDownload Raw Back to root
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  "batch_norm_eps": 1e-05,51  "box_noise_scale": 1.0,52  "d_model": 256,53  "decoder_activation_function": "relu",54  "decoder_attention_heads": 8,55  "decoder_ffn_dim": 1024,56  "decoder_in_channels": [57    256,58    256,59    25660  ],61  "decoder_layers": 6,62  "decoder_method": "default",63  "decoder_n_levels": 3,64  "decoder_n_points": 4,65  "decoder_offset_scale": 0.5,66  "disable_custom_kernels": true,67  "dropout": 0.0,68  "dtype": "float32",69  "encode_proj_layers": [70    271  ],72  "encoder_activation_function": "gelu",73  "encoder_attention_heads": 8,74  "encoder_ffn_dim": 1024,75  "encoder_hidden_dim": 256,76  "encoder_in_channels": [77    512,78    1024,79    204880  ],81  "encoder_layers": 1,82  "eos_coefficient": 0.0001,83  "eval_size": null,84  "feat_strides": [85    8,86    16,87    3288  ],89  "focal_loss_alpha": 0.75,90  "focal_loss_gamma": 2.0,91  "freeze_backbone_batch_norms": true,92  "hidden_expansion": 1.0,93  "id2label": {94    "0": "bin",95    "1": "hand",96    "2": "not_bin",97    "3": "not_hand",98    "4": "not_trash",99    "5": "trash",100    "6": "trash_arm"101  },102  "initializer_bias_prior_prob": null,103  "initializer_range": 0.01,104  "is_encoder_decoder": true,105  "label2id": {106    "bin": 0,107    "hand": 1,108    "not_bin": 2,109    "not_hand": 3,110    "not_trash": 4,111    "trash": 5,112    "trash_arm": 6113  },114  "label_noise_ratio": 0.5,115  "layer_norm_eps": 1e-05,116  "learn_initial_query": false,117  "matcher_alpha": 0.25,118  "matcher_bbox_cost": 5.0,119  "matcher_class_cost": 2.0,120  "matcher_gamma": 2.0,121  "matcher_giou_cost": 2.0,122  "model_type": "rt_detr_v2",123  "normalize_before": false,124  "num_denoising": 100,125  "num_feature_levels": 3,126  "num_queries": 300,127  "positional_encoding_temperature": 10000,128  "tie_word_embeddings": true,129  "transformers_version": "5.16.1",130  "use_focal_loss": true,131  "weight_loss_bbox": 5.0,132  "weight_loss_giou": 2.0,133  "weight_loss_vfl": 1.0,134  "with_box_refine": true135}136