CoolFace
Modelpublic

docling-project/DocumentFigureClassifier-v2.5

sourceHugging Facemitupdated 6mo agoView on Hugging Face
3likes84kdownloads
config.json146 linesDownload Raw Back to root
1{2  "architectures": [3    "EfficientNetForImageClassification"4  ],5  "batch_norm_eps": 0.001,6  "batch_norm_momentum": 0.99,7  "depth_coefficient": 1.0,8  "depth_divisor": 8,9  "depthwise_padding": [],10  "drop_connect_rate": 0.2,11  "dropout_rate": 0.2,12  "dtype": "float32",13  "expand_ratios": [14    1,15    6,16    6,17    6,18    6,19    6,20    621  ],22  "hidden_act": "swish",23  "hidden_dim": 1280,24  "id2label": {25    "0": "logo",26    "1": "photograph",27    "10": "signature",28    "11": "screenshot_from_manual",29    "12": "geographical_map",30    "13": "pie_chart",31    "14": "page_thumbnail",32    "15": "stamp",33    "16": "music",34    "17": "calendar",35    "18": "qr_code",36    "19": "bar_code",37    "2": "icon",38    "20": "full_page_image",39    "21": "scatter_plot",40    "22": "chemistry_structure",41    "23": "topographical_map",42    "24": "crossword_puzzle",43    "25": "box_plot",44    "3": "engineering_drawing",45    "4": "line_chart",46    "5": "bar_chart",47    "6": "other",48    "7": "table",49    "8": "flow_chart",50    "9": "screenshot_from_computer"51  },52  "image_size": 224,53  "in_channels": [54    32,55    16,56    24,57    40,58    80,59    112,60    19261  ],62  "initializer_range": 0.02,63  "kernel_sizes": [64    3,65    3,66    5,67    3,68    5,69    5,70    371  ],72  "label2id": {73    "bar_chart": "5",74    "bar_code": "19",75    "box_plot": "25",76    "calendar": "17",77    "chemistry_structure": "22",78    "crossword_puzzle": "24",79    "engineering_drawing": "3",80    "flow_chart": "8",81    "full_page_image": "20",82    "geographical_map": "12",83    "icon": "2",84    "line_chart": "4",85    "logo": "0",86    "music": "16",87    "other": "6",88    "page_thumbnail": "14",89    "photograph": "1",90    "pie_chart": "13",91    "qr_code": "18",92    "scatter_plot": "21",93    "screenshot_from_computer": "9",94    "screenshot_from_manual": "11",95    "signature": "10",96    "stamp": "15",97    "table": "7",98    "topographical_map": "23"99  },100  "model_type": "efficientnet",101  "num_block_repeats": [102    1,103    2,104    2,105    3,106    3,107    4,108    1109  ],110  "num_channels": 3,111  "num_hidden_layers": 64,112  "out_channels": [113    16,114    24,115    40,116    80,117    112,118    192,119    320120  ],121  "out_features": null,122  "pooling_type": "mean",123  "squeeze_expansion_ratio": 0.25,124  "stage_names": [125    "stem",126    "stage1",127    "stage2",128    "stage3",129    "stage4",130    "stage5",131    "stage6",132    "stage7"133  ],134  "strides": [135    1,136    2,137    2,138    2,139    1,140    2,141    1142  ],143  "transformers_version": "4.57.3",144  "width_coefficient": 1.0145}146