2 writes to PredictedLabelsBuffer
Microsoft.ML.TorchSharp (2)
AutoFormerV2\ObjectDetectionTrainer.cs (2)
947
PredictedLabelsBuffer
= default;
973
ImageUtils.Postprocess(imageTensor, pred, score, box, out outputCache.
PredictedLabelsBuffer
, out outputCache.ScoresBuffer, out outputCache.BoxBuffer, _parent.Options.ScoreThreshold, _parent.Options.IOUThreshold);
3 references to PredictedLabelsBuffer
Microsoft.ML.TorchSharp (3)
AutoFormerV2\ObjectDetectionTrainer.cs (3)
840
var editor = VBufferEditor.Create(ref dst, outputCacher.
PredictedLabelsBuffer
.Length);
842
for (var i = 0; i < outputCacher.
PredictedLabelsBuffer
.Length; i++)
844
editor.Values[i] = outputCacher.
PredictedLabelsBuffer
[i];