From f902d58c96e174d2f1c0fd0945f4a1867b2f3826 Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Fri, 14 Oct 2022 09:14:25 +0000 Subject: [PATCH] Upload TableTransformerForObjectDetection --- config.json | 51 +++++++++++++++++++++++++++++++++++++++++++++++ pytorch_model.bin | 3 +++ 2 files changed, 54 insertions(+) create mode 100644 config.json create mode 100644 pytorch_model.bin diff --git a/config.json b/config.json new file mode 100644 index 0000000..3e86a23 --- /dev/null +++ b/config.json @@ -0,0 +1,51 @@ +{ + "activation_dropout": 0.0, + "activation_function": "relu", + "architectures": [ + "TableTransformerForObjectDetection" + ], + "attention_dropout": 0.0, + "auxiliary_loss": false, + "backbone": "resnet18", + "bbox_cost": 5, + "bbox_loss_coefficient": 5, + "ce_loss_coefficient": 1, + "class_cost": 1, + "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, + "eos_coefficient": 0.4, + "giou_cost": 2, + "giou_loss_coefficient": 2, + "id2label": { + "0": "table", + "1": "table rotated" + }, + "init_std": 0.02, + "init_xavier_std": 1.0, + "is_encoder_decoder": true, + "label2id": { + "table": 0, + "table rotated": 1 + }, + "mask_loss_coefficient": 1, + "max_position_embeddings": 1024, + "model_type": "table-transformer", + "num_channels": 3, + "num_hidden_layers": 6, + "num_queries": 15, + "position_embedding_type": "sine", + "scale_embedding": false, + "torch_dtype": "float32", + "transformers_version": "4.24.0.dev0", + "use_pretrained_backbone": true +} diff --git a/pytorch_model.bin b/pytorch_model.bin new file mode 100644 index 0000000..d550834 --- /dev/null +++ b/pytorch_model.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:1d9babb11711211dd9fea2f477b7ba11f3656623505b07f9e30bfd41f143a9c7 +size 115393245