(label, pred)
| 246 | |
| 247 | @staticmethod |
| 248 | def disparity(label, pred): |
| 249 | label = (label > 0.5).astype(np.float32) |
| 250 | pred = (pred > 0.5).astype(np.float32) |
| 251 | disparity = label - pred |
| 252 | |
| 253 | pos_disparity = (disparity > 0).astype(np.float32) |
| 254 | neg_disparity = (disparity < 0).astype(np.float32) |
| 255 | return [pos_disparity, neg_disparity] |
| 256 | |
| 257 | def _apply(self, label, pred): |
| 258 | return self.disparity(label, pred) |