| 109 | return ci, cj |
| 110 | |
| 111 | def second_derivative(input, ci, cj): |
| 112 | u = input |
| 113 | m = u.shape[2] |
| 114 | n = u.shape[3] |
| 115 | |
| 116 | cii_0 = (u[:, :, 1, :] + u[:, :, 0, :] - |
| 117 | 2 * u[:, :, 0, :]).unsqueeze(2) |
| 118 | cii_1 = u[:, :, 2:, :] + u[:, :, :-2, :] - 2 * u[:, :, 1:-1, :] |
| 119 | cii_2 = (u[:, :, -1, :] + u[:, :, -2, :] - |
| 120 | 2 * u[:, :, -1, :]).unsqueeze(2) |
| 121 | cii = torch.cat([cii_0, cii_1, cii_2], 2) |
| 122 | |
| 123 | cjj_0 = (u[:, :, :, 1] + u[:, :, :, 0] - |
| 124 | 2 * u[:, :, :, 0]).unsqueeze(3) |
| 125 | cjj_1 = u[:, :, :, 2:] + u[:, :, :, :-2] - 2 * u[:, :, :, 1:-1] |
| 126 | cjj_2 = (u[:, :, :, -1] + u[:, :, :, -2] - |
| 127 | 2 * u[:, :, :, -1]).unsqueeze(3) |
| 128 | |
| 129 | cjj = torch.cat([cjj_0, cjj_1, cjj_2], 3) |
| 130 | |
| 131 | cij_0 = ci[:, :, :, 1:n] |
| 132 | cij_1 = ci[:, :, :, -1].unsqueeze(3) |
| 133 | |
| 134 | cij_a = torch.cat([cij_0, cij_1], 3) |
| 135 | cij_2 = ci[:, :, :, 0].unsqueeze(3) |
| 136 | cij_3 = ci[:, :, :, 0:n - 1] |
| 137 | cij_b = torch.cat([cij_2, cij_3], 3) |
| 138 | cij = cij_a - cij_b |
| 139 | |
| 140 | return cii, cjj, cij |
| 141 | |
| 142 | def region(y_pred, y_true, u=1): |
| 143 | label = y_true.float() |