Open Xiaobaicai13 opened 9 months ago
为什么我的这里 pointer_pred = torch.sigmoid(output[0, 0, :, :]).data.cpu().numpy() dail_pred = torch.sigmoid(output[0, 1, :, :]).data.cpu().numpy() text_pred = torch.sigmoid(output[0, 2, :, :]).data.cpu().numpy() pointer_pred = (pointer_pred > 0.5).astype(np.uint8) dail_pred = (dail_pred > 0.5).astype(np.uint8) text_pred = (text_pred > 0.7).astype(np.uint8) 最后的结果全是0啊,然后后面就报错
为什么我的这里 pointer_pred = torch.sigmoid(output[0, 0, :, :]).data.cpu().numpy() dail_pred = torch.sigmoid(output[0, 1, :, :]).data.cpu().numpy() text_pred = torch.sigmoid(output[0, 2, :, :]).data.cpu().numpy() pointer_pred = (pointer_pred > 0.5).astype(np.uint8) dail_pred = (dail_pred > 0.5).astype(np.uint8) text_pred = (text_pred > 0.7).astype(np.uint8) 最后的结果全是0啊,然后后面就报错