Fixed a bug in the unit test for image classification

Harshitha Parnandi Venkata · Harshitha Parnandi Venkata · commit b468adbd7d3b · 2019-10-01T21:19:57.000-07:00
diff --git a/src/Microsoft.ML.Dnn/ImageClassificationTransform.cs b/src/Microsoft.ML.Dnn/ImageClassificationTransform.cs
@@ -204,6 +204,7 @@ private static Tensor EncodeByteAsString(VBuffer<byte> buffer)
                 fixed (byte* src = buffer.GetValues())
                     c_api.TF_StringEncode(src, (UIntPtr)length, (sbyte*)(tensor + sizeof(Int64)), size, status);
             }
+
             status.Check(true);
             status.Dispose();
             return new Tensor(handle);
diff --git a/test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs b/test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs
@@ -1242,11 +1242,11 @@ public void TensorFlowImageClassification()
             IDataView trainDataset = trainTestData.TrainSet;
             IDataView testDataset = trainTestData.TestSet;
 
-            var validationSet = mlContext.Transforms.LoadImages("Image", fullImagesetFolderPath, true, "ImagePath") // true indicates we want the image as a VBuffer<byte>
+            var validationSet = mlContext.Transforms.LoadImages("Image", fullImagesetFolderPath, false, "ImagePath") // false indicates we want the image as a VBuffer<byte>
                     .Fit(testDataset)
                     .Transform(testDataset);
 
-            var pipeline = mlContext.Transforms.LoadImages("Image", fullImagesetFolderPath, true, "ImagePath") // true indicates we want the image as a VBuffer<byte>
+            var pipeline = mlContext.Transforms.LoadImages("Image", fullImagesetFolderPath, false, "ImagePath") // false indicates we want the image as a VBuffer<byte>
                 .Append(mlContext.Model.ImageClassification(
                     "Image", "Label",
                     // Just by changing/selecting InceptionV3 here instead of 

Original file line number	Diff line number	Diff line change
`@@ -204,6 +204,7 @@ private static Tensor EncodeByteAsString(VBuffer<byte> buffer)`
`204`	`204`	`fixed (byte* src = buffer.GetValues())`
`205`	`205`	`c_api.TF_StringEncode(src, (UIntPtr)length, (sbyte*)(tensor + sizeof(Int64)), size, status);`
`206`	`206`	`}`
	`207`	`+`
`207`	`208`	`status.Check(true);`
`208`	`209`	`status.Dispose();`
`209`	`210`	`return new Tensor(handle);`