Skip to content

Commit 9bf504d

Browse files
committed
Reorder function params
1 parent 2a2d130 commit 9bf504d

File tree

5 files changed

+7
-7
lines changed

5 files changed

+7
-7
lines changed

TensorStack.Common/ModelParameters.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,9 +315,9 @@ public void AddScalarInput(byte value)
315315
/// Adds the ImageTensor input.
316316
/// </summary>
317317
/// <param name="value">The value.</param>
318-
/// <param name="normalization">The normalization.</param>
319318
/// <param name="channels">The channels.</param>
320-
public void AddImageInput(ImageTensor value, Normalization normalization, int channels = 3)
319+
/// <param name="normalization">The normalization.</param>
320+
public void AddImageInput(ImageTensor value, int channels, Normalization normalization)
321321
{
322322
AddInput(value.GetChannels(channels), normalization);
323323
}

TensorStack.Extractors/Pipelines/BackgroundPipeline.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ private async Task<ImageTensor> ExtractBackgroundInternalAsync(BackgroundMode ba
9393
var outputBuffer = metadata.Outputs[0].Value.Dimensions.Length == 4 ? outputShape : outputShape[1..];
9494
using (var modelParameters = new ModelParameters(metadata, cancellationToken))
9595
{
96-
modelParameters.AddImageInput(inputTensor, _model.Normalization, _model.Channels);
96+
modelParameters.AddImageInput(inputTensor, _model.Channels, _model.Normalization);
9797
modelParameters.AddOutput(outputBuffer);
9898
using (var results = await _model.RunInferenceAsync(modelParameters))
9999
{

TensorStack.Extractors/Pipelines/ExtractorPipeline.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ private async Task<ImageTensor> ExecuteExtractorAsync(ImageTensor imageInput, Ca
176176
var outputBuffer = metadata.Outputs[0].Value.Dimensions.Length == 4 ? outputShape : outputShape[1..];
177177
using (var modelParameters = new ModelParameters(metadata, cancellationToken))
178178
{
179-
modelParameters.AddImageInput(inputTensor, _model.Normalization, _model.Channels);
179+
modelParameters.AddImageInput(inputTensor, _model.Channels, _model.Normalization);
180180
modelParameters.AddOutput(outputBuffer);
181181

182182
var results = _model.IsDynamicOutput

TensorStack.Upscaler/Pipelines/UpscalePipeline.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ private async Task<ImageTensor> ExecuteUpscaleAsync(ImageTensor imageTensor, Can
159159

160160
using (var modelParameters = new ModelParameters(metadata, cancellationToken))
161161
{
162-
modelParameters.AddImageInput(imageTensor, _model.Normalization, _model.Channels);
162+
modelParameters.AddImageInput(imageTensor, _model.Channels, _model.Normalization);
163163
modelParameters.AddOutput([1, _model.Channels, imageTensor.Height * _model.ScaleFactor, imageTensor.Width * _model.ScaleFactor]);
164164
using (var results = await _model.RunInferenceAsync(modelParameters))
165165
{

TensorStack.Video/Pipelines/InterpolationPipeline.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,8 @@ private async Task<ImageTensor> RunInterpolationAsync(ImageTensor frameTensor, I
156156
using (var modelParameters = new ModelParameters(metadata, cancellationToken))
157157
{
158158
// Inputs
159-
modelParameters.AddImageInput(previousFrameTensor, _model.Normalization, _model.Channels);
160-
modelParameters.AddImageInput(frameTensor, _model.Normalization, _model.Channels);
159+
modelParameters.AddImageInput(previousFrameTensor, _model.Channels, _model.Normalization);
160+
modelParameters.AddImageInput(frameTensor, _model.Channels, _model.Normalization);
161161
modelParameters.AddScalarInput(timestep);
162162

163163
// Outputs

0 commit comments

Comments
 (0)