Skip to content

Commit

Permalink
ProcessMask parameters wraping
Browse files Browse the repository at this point in the history
  • Loading branch information
dme-compunet committed Sep 13, 2023
1 parent 6d993f9 commit ae94183
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion Source/YoloV8/Parsers/SegmentationOutputParser.cs
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,13 @@ public IReadOnlyList<ISegmentationBoundingBox> Parse(IReadOnlyList<Tensor<float>
});
}

private static IMask ProcessMask(Tensor<float> maskPrototypes, ReadOnlySpan<float> maskWeights, Rectangle bounds, Size originSize, Size modelSize, int xPadding, int yPadding)
private static IMask ProcessMask(Tensor<float> maskPrototypes,
ReadOnlySpan<float> maskWeights,
Rectangle bounds,
Size originSize,
Size modelSize,
int xPadding,
int yPadding)
{
var maskChannels = maskPrototypes.Dimensions[1];
var maskHeight = maskPrototypes.Dimensions[2];
Expand Down

0 comments on commit ae94183

Please sign in to comment.