getImagePredictionList method

Future<List<double?>?> getImagePredictionList(
  1. Uint8List imageAsBytes, {
  2. List<double> mean = TORCHVISION_NORM_MEAN_RGB,
  3. List<double> std = TORCHVISION_NORM_STD_RGB,
})

Runs image classification and returns raw prediction scores.

Returns the raw output scores from the neural network before applying softmax or argmax operations.

@param imageAsBytes The raw image bytes @param mean Optional normalization mean values (default: ImageNet means) @param std Optional normalization std values (default: ImageNet stds) @return A Future that completes with a list of raw prediction scores

Implementation

Future<List<double?>?> getImagePredictionList(Uint8List imageAsBytes,
    {List<double> mean = TORCHVISION_NORM_MEAN_RGB,
    List<double> std = TORCHVISION_NORM_STD_RGB}) async {
  // Assert mean std
  assert(mean.length == 3, "Mean should have size of 3");
  assert(std.length == 3, "STD should have size of 3");
  final List<double?> prediction = await ModelApi().getImagePredictionList(
      _index, imageAsBytes, null, null, null, mean, std);
  return prediction;
}