Only pad if pad_size is defined

huggingface · Sep 20, 2023 · e74844f · e74844f
1 parent 5824968
commit e74844f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/processors.js b/src/processors.js
@@ -319,7 +319,7 @@ export class ImageFeatureExtractor extends FeatureExtractor {
         let reshaped_input_size = [image.height, image.width];
 
         // TODO is it okay to pad before rescaling/normalizing?
-        if (this.do_pad) {
+        if (this.do_pad && this.pad_size) {
             let left = 0;
             let right = this.pad_size.width - image.width;
             let top = 0;