shuyansy / Detect-and-read-meters

This is the first released system towards complex meters` detection and recognition, which is implemented by computer vision techniques.
MIT License
110 stars 15 forks source link

检测问题 #15

Open Xiaobaicai13 opened 9 months ago

Xiaobaicai13 commented 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啊,然后后面就报错