Skip to content

Commit 0b1554b

Browse files
committed
fix build from merge
1 parent deded50 commit 0b1554b

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

test/Microsoft.ML.Tests/ScenariosWithDirectInstantiation/TensorflowTests.cs

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ public void TensorFlowTransformObjectDetectionTest()
238238
var images = new ImageLoadingTransformer(mlContext, imageFolder, ("ImageReal", "ImagePath")).Transform(data);
239239
var cropped = new ImageResizingTransformer(mlContext, "ImageCropped", 32, 32, "ImageReal").Transform(images);
240240

241-
var pixels = new ImagePixelExtractorTransformer(mlContext, "image_tensor", "ImageCropped", asFloat: false).Transform(cropped);
241+
var pixels = mlContext.Transforms.ExtractPixels("image_tensor", "ImageCropped", asFloat: false).Fit(cropped).Transform(cropped);
242242
var tf = mlContext.Transforms.ScoreTensorFlowModel(modelLocation,
243243
new[] { "detection_boxes", "detection_scores", "num_detections", "detection_classes" }, new[] { "image_tensor" }).Fit(pixels).Transform(pixels);
244244

@@ -276,9 +276,9 @@ public void TensorFlowTransformInceptionTest()
276276
var dataFile = GetDataPath("images/images.tsv");
277277
var imageFolder = Path.GetDirectoryName(dataFile);
278278
var data = mlContext.CreateLoader("Text{col=ImagePath:TX:0 col=Name:TX:1}", new MultiFileSource(dataFile));
279-
var images = new ImageLoaderTransformer(mlContext, imageFolder, ("ImageReal", "ImagePath")).Transform(data);
280-
var cropped = new ImageResizerTransformer(mlContext, "ImageCropped", 224, 224 , "ImageReal").Transform(images);
281-
var pixels = new ImagePixelExtractorTransformer(mlContext, "input","ImageCropped").Transform(cropped);
279+
var images = mlContext.Transforms.LoadImages(imageFolder, ("ImageReal", "ImagePath")).Fit(data).Transform(data);
280+
var cropped = mlContext.Transforms.Resize("ImageCropped", 224, 224 , "ImageReal").Fit(images).Transform(images);
281+
var pixels = mlContext.Transforms.ExtractPixels("input","ImageCropped").Fit(cropped).Transform(cropped);
282282
var tf = mlContext.Transforms.ScoreTensorFlowModel(modelLocation, "softmax2_pre_activation", "input").Fit(pixels).Transform(pixels);
283283

284284
tf.Schema.TryGetColumnIndex("input", out int input);
@@ -781,9 +781,9 @@ public void TensorFlowTransformCifarSavedModel()
781781
new TextLoader.Column("Name", DataKind.TX, 1),
782782
}
783783
);
784-
var images = new ImageLoaderTransformer(mlContext, imageFolder, ("ImageReal", "ImagePath")).Transform(data);
785-
var cropped = new ImageResizerTransformer(mlContext, "ImageCropped", imageWidth, imageHeight, "ImageReal").Transform(images);
786-
var pixels = new ImagePixelExtractorTransformer(mlContext, "Input", "ImageCropped", interleave: true).Transform(cropped);
784+
var images = mlContext.Transforms.LoadImages(imageFolder, ("ImageReal", "ImagePath")).Fit(data).Transform(data);
785+
var cropped = mlContext.Transforms.Resize("ImageCropped", imageWidth, imageHeight, "ImageReal").Fit(images).Transform(images);
786+
var pixels = mlContext.Transforms.ExtractPixels("Input", "ImageCropped", interleave: true).Fit(cropped).Transform(cropped);
787787
IDataView trans = mlContext.Transforms.ScoreTensorFlowModel(tensorFlowModel, "Output", "Input").Fit(pixels).Transform(pixels);
788788

789789
trans.Schema.TryGetColumnIndex("Output", out int output);

0 commit comments

Comments
 (0)