Fixed preprocessing rotation bug. Fixes #20

sampepose · Apr 15, 2018 · 18f8708 · 18f8708
1 parent 0fc1785
commit 18f8708
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/preprocessing/kernels/augmentation_base.cc b/src/ops/preprocessing/kernels/augmentation_base.cc
@@ -339,7 +339,7 @@ void AugmentationLayerBase::generate_valid_spatial_coeffs(
 
         // rotate
         x2 = cos(coeff.angle()) * x1 - sin(coeff.angle()) * y1;
-        y2 = sin(coeff.angle()) * x1 + sin(coeff.angle()) * y1;
+        y2 = sin(coeff.angle()) * x1 + cos(coeff.angle()) * y1;
 
         // translate
         x2 = x2 + coeff.dx() * out_width;