Browse code
fix bug in curvature_loss
Bas Nijholt authored on 15/02/2019 15:00:13
• GitHub committed on 15/02/2019 15:00:13
Showing 1 changed files
| ... | ... |
@@ -129,7 +129,7 @@ def curvature_loss_function(area_factor=1, euclid_factor=0.02, horizontal_factor |
| 129 | 129 |
@uses_nth_neighbors(1) |
| 130 | 130 |
def curvature_loss(xs, ys): |
| 131 | 131 |
xs_middle = xs[1:3] |
| 132 |
- ys_middle = xs[1:3] |
|
| 132 |
+ ys_middle = ys[1:3] |
|
| 133 | 133 |
|
| 134 | 134 |
triangle_loss_ = triangle_loss(xs, ys) |
| 135 | 135 |
default_loss_ = default_loss(xs_middle, ys_middle) |