PSENet icon indicating copy to clipboard operation
PSENet copied to clipboard

这里结果不对吧? kernels大于0判断,为啥去大于0的,这里的kernels阈值怎么会是0呢

Open cqray1990 opened this issue 3 years ago • 1 comments

score = torch.sigmoid(out[:, 0, :, :]) # out = (torch.sign(out - 1) + 1) / 2 # 0 1 # # text_mask = out[:, 0, :, :] # kernels = out[:, 1:cfg.test_cfg.kernel_num, :, :] * text_mask

    kernels = out[:, :cfg.test_cfg.kernel_num, :, :] > 0
    text_mask = kernels[:, :1, :, :]
    kernels[:, 1:, :, :] = kernels[:, 1:, :, :] * text_mask

cqray1990 avatar Jul 06 '22 09:07 cqray1990

No module named 'models.post_processing.pa.pa',这个问题您遇到过吗?

zeng-cy avatar Nov 19 '22 14:11 zeng-cy