PSENet
PSENet copied to clipboard
这里结果不对吧? kernels大于0判断,为啥去大于0的,这里的kernels阈值怎么会是0呢
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
No module named 'models.post_processing.pa.pa',这个问题您遇到过吗?