model_artifacts/tableformer/tm_config.json

{
  "dataset": {
    "type": "PTN_prepared",
    "name": "PubTabNet_300_100_512",
    "raw_data_dir": "./tests/test_data/ccs_api/model/",
    "load_cells": true,
    "bbox_format": "5plet",
    "resized_image": 448,
    "keep_AR": false,
    "up_scaling_enabled": true,
    "down_scaling_enabled": true,
    "padding_mode": "null",
    "padding_color": [
      0,
      0,
      0
    ],
    "image_normalization": {
      "state": true,
      "mean": [
        0.94247851,
        0.94254675,
        0.94292611
      ],
      "std": [
        0.17910956,
        0.17940403,
        0.17931663
      ]
    },
    "color_jitter": true,
    "rand_crop": true,
    "rand_pad": true,
    "image_grayscale": false
  },
  "model": {
    "type": "TableModel04_rs",
    "name": "14_128_256_4_true",
    "backbone": "resnet18",
    "enc_image_size": 28,
    "tag_embed_dim": 16,
    "hidden_dim": 512,
    "tag_decoder_dim": 512,
    "bbox_embed_dim": 256,
    "tag_attention_dim": 256,
    "bbox_attention_dim": 512,
    "enc_layers": 4,
    "dec_layers": 2,
    "nheads": 8,
    "dropout": 0.1,
    "bbox_classes": 2
  },
  "train": {
    "bbox": true
  },
  "predict": {
    "max_steps": 1024,
    "beam_size": 5,
    "bbox": true,
    "pdf_cell_iou_thres": 0.05,
    "padding": false,
    "padding_size": 50,
    "disable_post_process": false,
    "profiling": false,
    "device_mode": "auto"
  },
  "debug": {
    "save_debug_images": false
  },
  "dataset_wordmap": {
    "word_map_tag": {
      "<pad>": 0,
      "<unk>": 1,
      "<start>": 2,
      "<end>": 3,
      "ecel": 4,
      "fcel": 5,
      "lcel": 6,
      "ucel": 7,
      "xcel": 8,
      "nl": 9,
      "ched": 10,
      "rhed": 11,
      "srow": 12
    },
    "word_map_cell": {
      " ": 13,
      "!": 179,
      "\"": 126,
      "#": 101,
      "$": 119,
      "%": 18,
      "&": 114,
      "'": 108,
      "(": 29,
      ")": 32,
      "*": 26,
      "+": 97,
      ",": 71,
      "-": 63,
      ".": 34,
      "/": 66,
      "0": 33,
      "1": 36,
      "2": 43,
      "3": 41,
      "4": 45,
      "5": 17,
      "6": 37,
      "7": 35,
      "8": 40,
      "9": 16,
      ":": 88,
      ";": 92,
      "<": 73,
      "</b>": 9,
      "</i>": 23,
      "</overline>": 219,
      "</strike>": 233,
      "</sub>": 94,
      "</sup>": 77,
      "</underline>": 151,
      "<b>": 1,
      "<end>": 280,
      "<i>": 21,
      "<overline>": 218,
      "<pad>": 0,
      "<start>": 279,
      "<strike>": 232,
      "<sub>": 93,
      "<sup>": 75,
      "<underline>": 150,
      "<unk>": 278,
      "=": 99,
      ">": 39,
      "?": 96,
      "@": 125,
      "A": 27,
      "B": 86,
      "C": 19,
      "D": 57,
      "E": 64,
      "F": 47,
      "G": 44,
      "H": 10,
      "I": 20,
      "J": 80,
      "K": 81,
      "L": 52,
      "M": 46,
      "N": 69,
      "O": 65,
      "P": 62,
      "Q": 59,
      "R": 60,
      "S": 58,
      "T": 48,
      "U": 55,
      "V": 2,
      "W": 83,
      "X": 104,
      "Y": 89,
      "Z": 113,
      "[": 70,
      "\\": 165,
      "]": 72,
      "^": 132,
      "_": 84,
      "`": 196,
      "a": 3,
      "b": 6,
      "c": 54,
      "d": 12,
      "e": 8,
      "f": 50,
      "g": 28,
      "h": 56,
      "i": 5,
      "j": 82,
      "k": 95,
      "l": 7,
      "m": 30,
      "n": 31,
      "o": 15,
      "p": 22,
      "q": 67,
      "r": 4,
      "s": 51,
      "t": 14,
      "u": 25,
      "v": 24,
      "w": 53,
      "x": 61,
      "y": 49,
      "z": 11,
      "{": 158,
      "|": 139,
      "}": 159,
      "~": 147,
      "\u00a2": 203,
      "\u00a3": 162,
      "\u00a4": 220,
      "\u00a5": 176,
      "\u00a7": 142,
      "\u00a9": 268,
      "\u00ab": 239,
      "\u00ad": 275,
      "\u00ae": 130,
      "\u00b0": 100,
      "\u00b1": 79,
      "\u00b6": 171,
      "\u00b7": 137,
      "\u00bb": 240,
      "\u00d7": 118,
      "\u00d8": 192,
      "\u00df": 197,
      "\u00e6": 261,
      "\u00f7": 225,
      "\u00f8": 163,
      "\u0131": 242,
      "\u0142": 267,
      "\u01c2": 211,
      "\u025b": 223,
      "\u02b9": 248,
      "\u02c2": 195,
      "\u02c3": 208,
      "\u02c6": 253,
      "\u0300": 209,
      "\u0301": 131,
      "\u0302": 138,
      "\u0303": 156,
      "\u0304": 152,
      "\u0306": 222,
      "\u0307": 247,
      "\u0308": 103,
      "\u030a": 102,
      "\u030c": 254,
      "\u0327": 155,
      "\u0328": 269,
      "\u0338": 170,
      "\u0391": 173,
      "\u0392": 169,
      "\u0393": 180,
      "\u0394": 85,
      "\u0398": 243,
      "\u0399": 271,
      "\u039b": 272,
      "\u03a0": 213,
      "\u03a3": 185,
      "\u03a6": 148,
      "\u03a7": 212,
      "\u03a8": 141,
      "\u03a9": 161,
      "\u03b1": 90,
      "\u03b2": 107,
      "\u03b3": 110,
      "\u03b4": 153,
      "\u03b5": 166,
      "\u03b6": 178,
      "\u03b7": 146,
      "\u03b8": 186,
      "\u03b9": 229,
      "\u03ba": 164,
      "\u03bb": 91,
      "\u03bc": 78,
      "\u03bd": 230,
      "\u03be": 244,
      "\u03c0": 127,
      "\u03c1": 149,
      "\u03c3": 116,
      "\u03c4": 198,
      "\u03c5": 189,
      "\u03c6": 140,
      "\u03c7": 124,
      "\u03c8": 216,
      "\u03c9": 167,
      "\u0410": 273,
      "\u0421": 194,
      "\u115f": 217,
      "\u200b": 265,
      "\u2010": 117,
      "\u2012": 135,
      "\u2013": 42,
      "\u2014": 106,
      "\u2015": 228,
      "\u2016": 259,
      "\u2018": 123,
      "\u2019": 121,
      "\u201c": 87,
      "\u201d": 115,
      "\u201e": 245,
      "\u2020": 109,
      "\u2021": 129,
      "\u2022": 128,
      "\u2028": 190,
      "\u2030": 154,
      "\u2032": 68,
      "\u203b": 224,
      "\u2044": 188,
      "\u204e": 199,
      "\u2061": 200,
      "\u20ac": 184,
      "\u2190": 202,
      "\u2191": 112,
      "\u2192": 120,
      "\u2193": 111,
      "\u2194": 183,
      "\u21d1": 266,
      "\u21d2": 264,
      "\u21d3": 255,
      "\u2205": 215,
      "\u2206": 175,
      "\u2208": 262,
      "\u2211": 160,
      "\u2212": 76,
      "\u2216": 206,
      "\u2217": 105,
      "\u2218": 246,
      "\u2219": 236,
      "\u221a": 187,
      "\u221e": 207,
      "\u2223": 260,
      "\u2225": 193,
      "\u2227": 182,
      "\u2229": 256,
      "\u222b": 258,
      "\u223c": 98,
      "\u2248": 210,
      "\u2264": 38,
      "\u2265": 74,
      "\u2266": 214,
      "\u2267": 181,
      "\u2295": 263,
      "\u22c5": 174,
      "\u22c6": 191,
      "\u22ee": 277,
      "\u22ef": 270,
      "\u2500": 205,
      "\u2551": 231,
      "\u25a0": 250,
      "\u25a1": 177,
      "\u25aa": 145,
      "\u25b2": 136,
      "\u25b3": 143,
      "\u25bc": 251,
      "\u25c6": 226,
      "\u25ca": 235,
      "\u25cb": 227,
      "\u25cf": 172,
      "\u25e6": 274,
      "\u2605": 204,
      "\u2606": 144,
      "\u2640": 133,
      "\u2642": 134,
      "\u2663": 252,
      "\u2666": 157,
      "\u266f": 221,
      "\u2713": 122,
      "\u2714": 249,
      "\u2717": 201,
      "\u2794": 168,
      "\u27a2": 276,
      "\u2a7d": 234,
      "\u2a7e": 241,
      "\u3008": 237,
      "\u3009": 238,
      "\ufeff": 257
    }
  }
}