(self, y_pred, y_true)
| 381 | return elastica |
| 382 | |
| 383 | def forward(self, y_pred, y_true): |
| 384 | loss = self.region(y_pred, y_true, u=self.miu) + \ |
| 385 | self.elastica(y_pred, a=self.alpha, b=self.beta) |
| 386 | return loss |
| 387 | |
| 388 | |
| 389 | class FastACELoss3D(nn.Module): |