search

D641593 / MixNet

MIT License
66 stars 9 forks source link

error for ICDAR-ArT model to load weights #16

Open world2vec opened 9 months ago

world2vec commented 9 months ago

Hi, Thanks for your great work! It is OK to load model MSRA-TD500 but when I load the weight for ICDAR-ArT, it errored:

RuntimeError: Error(s) in loading state_dict for TextNet:
    Unexpected key(s) in state_dict: "embed_head.0.weight", "embed_head.0.bias".

below is the code used:

cfg.resume = False
cfg.embed = False
cfg.onlybackbone = False
cfg.mid = True
cfg.dis_threshold = 0.4
cfg.cls_threshold = 0.8
cfg.test_size = [960, 2880]
cfg.device='cpu'
model = TextNet(is_training=False, backbone='FSNet_M')
model.load_model('../data/pretrained/mixnet/MixNet_FSNet_M_160.pth')
lerndeep commented 8 months ago

@world2vec I also faced same error. have you solved this error? could you please let me know your MSRA-TD500 evaluation configuration? I face same error during evalutaion on msra-td500 dataset too. Thank you