Apppublic

fendiprime/dit-document-layout-analysis

sourceHugging Faceupdated 3y agoView on Hugging Face

0likes

cascade_dit_base.yml21 linesDownload Raw Back to root

1_BASE_: "Base-RCNN-FPN.yml"2MODEL:3  PIXEL_MEAN: [ 127.5, 127.5, 127.5 ]4  PIXEL_STD: [ 127.5, 127.5, 127.5 ]5  WEIGHTS: "https://layoutlm.blob.core.windows.net/dit/dit-pts/dit-base-224-p16-500k-62d53a.pth"6  VIT:7    NAME: "dit_base_patch16"8  ROI_HEADS:9    NAME: CascadeROIHeads10  ROI_BOX_HEAD:11    CLS_AGNOSTIC_BBOX_REG: True12  RPN:13    POST_NMS_TOPK_TRAIN: 200014SOLVER:15  WARMUP_ITERS: 100016  IMS_PER_BATCH: 1617  MAX_ITER: 6000018  CHECKPOINT_PERIOD: 200019TEST:20  EVAL_PERIOD: 200021