Skip to content

Commit 283982e

Browse files
committed
PR feedback.
1 parent 18629ea commit 283982e

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

src/Microsoft.ML.DnnImageFeaturizer.AlexNet/AlexNetExtension.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@
66
using System.IO;
77
using Microsoft.ML.Data;
88
using Microsoft.ML.Runtime;
9+
using Microsoft.ML.Transforms;
10+
using Microsoft.ML.Transforms.Onnx;
911

10-
namespace Microsoft.ML.Transforms.Onnx
12+
namespace Microsoft.ML
1113
{
1214
/// <summary>
1315
/// This is an extension method to be used with the <see cref="DnnImageFeaturizerEstimator"/> in order to use a pretrained AlexNet model.

src/Microsoft.ML.DnnImageFeaturizer.ResNet101/ResNet101Extension.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@
66
using System.IO;
77
using Microsoft.ML.Data;
88
using Microsoft.ML.Runtime;
9+
using Microsoft.ML.Transforms;
10+
using Microsoft.ML.Transforms.Onnx;
911

10-
namespace Microsoft.ML.Transforms.Onnx
12+
namespace Microsoft.ML
1113
{
1214
/// <summary>
1315
/// This is an extension method to be used with the <see cref="DnnImageFeaturizerEstimator"/> in order to use a pretrained ResNet101 model.

src/Microsoft.ML.DnnImageFeaturizer.ResNet18/ResNet18Extension.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@
66
using System.IO;
77
using Microsoft.ML.Data;
88
using Microsoft.ML.Runtime;
9+
using Microsoft.ML.Transforms;
10+
using Microsoft.ML.Transforms.Onnx;
911

10-
namespace Microsoft.ML.Transforms.Onnx
12+
namespace Microsoft.ML
1113
{
1214
/// <summary>
1315
/// This is an extension method to be used with the <see cref="DnnImageFeaturizerEstimator"/> in order to use a pretrained ResNet18 model.

src/Microsoft.ML.DnnImageFeaturizer.ResNet50/ResNet50Extension.cs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,10 @@
66
using System.IO;
77
using Microsoft.ML.Data;
88
using Microsoft.ML.Runtime;
9+
using Microsoft.ML.Transforms;
10+
using Microsoft.ML.Transforms.Onnx;
911

10-
namespace Microsoft.ML.Transforms.Onnx
12+
namespace Microsoft.ML
1113
{
1214
/// <summary>
1315
/// This is an extension method to be used with the <see cref="DnnImageFeaturizerEstimator"/> in order to use a pretrained ResNet50 model.

0 commit comments

Comments
 (0)