upload trashify box detection model trained on purely synthetic data

README.md

@@ -0,0 +1,108 @@
+---
+license: apache-2.0
+base_model: microsoft/conditional-detr-resnet-50
+tags:
+- generated_from_trainer
+model-index:
+- name: trashify-box-detection-model-synthetic-data-only
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# trashify-box-detection-model-synthetic-data-only
+
+This model is a fine-tuned version of [microsoft/conditional-detr-resnet-50](https://huggingface.co/microsoft/conditional-detr-resnet-50) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.9981
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 0.0001
+- train_batch_size: 32
+- eval_batch_size: 32
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_ratio: 0.05
+- num_epochs: 50
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| 61.6636       | 1.0   | 83   | 4.7558          |
+| 3.6677        | 2.0   | 166  | 2.7089          |
+| 2.8053        | 3.0   | 249  | 2.4394          |
+| 2.5214        | 4.0   | 332  | 2.0961          |
+| 2.3314        | 5.0   | 415  | 1.9080          |
+| 2.1984        | 6.0   | 498  | 1.7241          |
+| 2.1242        | 7.0   | 581  | 1.6452          |
+| 2.0397        | 8.0   | 664  | 1.5851          |
+| 1.9768        | 9.0   | 747  | 1.6050          |
+| 1.9331        | 10.0  | 830  | 1.4683          |
+| 1.9034        | 11.0  | 913  | 1.4983          |
+| 1.8388        | 12.0  | 996  | 1.4433          |
+| 1.8016        | 13.0  | 1079 | 1.3840          |
+| 1.7536        | 14.0  | 1162 | 1.3470          |
+| 1.7232        | 15.0  | 1245 | 1.3208          |
+| 1.7181        | 16.0  | 1328 | 1.3386          |
+| 1.6767        | 17.0  | 1411 | 1.2681          |
+| 1.657         | 18.0  | 1494 | 1.2348          |
+| 1.6222        | 19.0  | 1577 | 1.2102          |
+| 1.5906        | 20.0  | 1660 | 1.1885          |
+| 1.546         | 21.0  | 1743 | 1.1645          |
+| 1.5524        | 22.0  | 1826 | 1.1788          |
+| 1.5119        | 23.0  | 1909 | 1.1606          |
+| 1.4963        | 24.0  | 1992 | 1.1689          |
+| 1.4932        | 25.0  | 2075 | 1.1395          |
+| 1.4651        | 26.0  | 2158 | 1.1246          |
+| 1.4358        | 27.0  | 2241 | 1.1250          |
+| 1.4294        | 28.0  | 2324 | 1.1090          |
+| 1.4277        | 29.0  | 2407 | 1.1183          |
+| 1.4061        | 30.0  | 2490 | 1.1300          |
+| 1.3885        | 31.0  | 2573 | 1.0943          |
+| 1.3731        | 32.0  | 2656 | 1.0680          |
+| 1.3811        | 33.0  | 2739 | 1.0820          |
+| 1.3436        | 34.0  | 2822 | 1.0585          |
+| 1.3314        | 35.0  | 2905 | 1.0574          |
+| 1.347         | 36.0  | 2988 | 1.0766          |
+| 1.3341        | 37.0  | 3071 | 1.0249          |
+| 1.3188        | 38.0  | 3154 | 1.0250          |
+| 1.3057        | 39.0  | 3237 | 1.0263          |
+| 1.2932        | 40.0  | 3320 | 1.0307          |
+| 1.2703        | 41.0  | 3403 | 1.0137          |
+| 1.286         | 42.0  | 3486 | 1.0072          |
+| 1.2864        | 43.0  | 3569 | 1.0097          |
+| 1.2493        | 44.0  | 3652 | 0.9997          |
+| 1.2511        | 45.0  | 3735 | 0.9994          |
+| 1.2428        | 46.0  | 3818 | 1.0030          |
+| 1.2306        | 47.0  | 3901 | 0.9964          |
+| 1.2258        | 48.0  | 3984 | 1.0002          |
+| 1.2307        | 49.0  | 4067 | 0.9977          |
+| 1.2251        | 50.0  | 4150 | 0.9981          |
+
+
+### Framework versions
+
+- Transformers 4.44.0
+- Pytorch 2.2.0+cu121
+- Datasets 2.20.0
+- Tokenizers 0.19.1

config.json

@@ -0,0 +1,71 @@
+{
+  "_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": "bin",
+    "1": "hand",
+    "2": "not_bin",
+    "3": "not_hand",
+    "4": "not_trash",
+    "5": "trash"
+  },
+  "init_std": 0.02,
+  "init_xavier_std": 1.0,
+  "is_encoder_decoder": true,
+  "label2id": {
+    "bin": 0,
+    "hand": 1,
+    "not_bin": 2,
+    "not_hand": 3,
+    "not_trash": 4,
+    "trash": 5
+  },
+  "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.44.0",
+  "use_pretrained_backbone": true,
+  "use_timm_backbone": true
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaeaa56157c65f7fc5b845f3e14329b0c378dabe54c8bd40bfd06619b1c62d91
+size 174080824

preprocessor_config.json

@@ -0,0 +1,29 @@
+{
+  "do_convert_annotations": true,
+  "do_normalize": true,
+  "do_pad": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "format": "coco_detection",
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "ConditionalDetrImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "pad_size": {
+    "height": 480,
+    "width": 480
+  },
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 480,
+    "width": 480
+  }
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a1af0f5f89b76bb6adbbc6e1a718eed986fcd2ad35bf19d3ec425ca7ea29d2e
+size 5240