diff --git a/src/processors.js b/src/processors.js index 4dcb4862f..a7df4cd2e 100644 --- a/src/processors.js +++ b/src/processors.js @@ -700,7 +700,7 @@ export class ImageFeatureExtractor extends FeatureExtractor { const pixel_values = stack(imageData.map(x => x.pixel_values), 0); return { - pixel_values: pixel_values, + pixel_values, // Original sizes of images original_sizes: imageData.map(x => x.original_size),