Merge pull request #1143 from dogvane/master

Oceania2018 · web-flow · commit 6dd15f7353db · 2023-07-10T14:42:02.000-05:00
fix flip_left_right run bug
diff --git a/src/TensorFlowNET.Core/Operations/image_ops_impl.cs b/src/TensorFlowNET.Core/Operations/image_ops_impl.cs
@@ -208,7 +208,7 @@ internal static Tensor _random_flip(Tensor image, int flip_index, int seed, stri
         }
 
         public static Tensor flip_left_right(Tensor image)
-            => _flip(image, 1, "flip_left_right");
+            => _flip(image, 0, "flip_left_right");
 
         public static Tensor flip_up_down(Tensor image)
             => _flip(image, 1, "flip_up_down");
@@ -226,7 +226,7 @@ internal static Tensor _flip(Tensor image, int flip_index, string scope_name)
                   }
                   else if (shape.ndim == 4)
                   {
-                      return gen_array_ops.reverse(image, ops.convert_to_tensor(new[] { flip_index + 1 }));
+                      return gen_array_ops.reverse_v2(image, ops.convert_to_tensor(new[] { (flip_index + 1) % 2 }));
                   }
                   else
                   {

Original file line number	Diff line number	Diff line change
`@@ -208,7 +208,7 @@ internal static Tensor _random_flip(Tensor image, int flip_index, int seed, stri`
`208`	`208`	`}`
`209`	`209`
`210`	`210`	`public static Tensor flip_left_right(Tensor image)`
`211`		`- => _flip(image, 1, "flip_left_right");`
	`211`	`+ => _flip(image, 0, "flip_left_right");`
`212`	`212`
`213`	`213`	`public static Tensor flip_up_down(Tensor image)`
`214`	`214`	`=> _flip(image, 1, "flip_up_down");`
`@@ -226,7 +226,7 @@ internal static Tensor _flip(Tensor image, int flip_index, string scope_name)`
`226`	`226`	`}`
`227`	`227`	`else if (shape.ndim == 4)`
`228`	`228`	`{`
`229`		`- return gen_array_ops.reverse(image, ops.convert_to_tensor(new[] { flip_index + 1 }));`
	`229`	`+ return gen_array_ops.reverse_v2(image, ops.convert_to_tensor(new[] { (flip_index + 1) % 2 }));`
`230`	`230`	`}`
`231`	`231`	`else`
`232`	`232`	`{`