WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库,支持常用的文字检测和识别算法

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

使用Transformer 要加载预训练权重 代码里的预训练权重在哪呢?

bowenliu1996 opened this issue · comments

    new_ckpt = OrderedDict()
    ckpt_path = './weights/upernet_swin_tiny_patch4_window7_512x512.pth'
    weights_dict = torch.load(ckpt_path)["state_dict"]

thanks