cdameron1024/trashify-object-detector
031
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 