Bugfix in resize_to_range to allow image with arbitrary number of channels.

PiperOrigin-RevId: 341573525

Bugfix in resize_to_range to allow image with arbitrary number of channels.
PiperOrigin-RevId: 341573525
f858af81 · A. Unique TensorFlower · TF Object Detection Team · 0b7646e2 · f858af81
Commit f858af81 authored Nov 10, 2020 by A. Unique TensorFlower Committed by TF Object Detection Team Nov 10, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

research/object_detection/core/preprocessor.py research/object_detection/core/preprocessor.py +1 -1

No files found.
--- a/research/object_detection/core/preprocessor.py
+++ b/research/object_detection/core/preprocessor.py
@@ -2937,7 +2937,7 @@ def resize_to_range(image,
              for i in range(len(channels))
          ],
          axis=2)
-      new_image.set_shape([max_dimension, max_dimension, 3])
+      new_image.set_shape([max_dimension, max_dimension, len(channels)])

    result = [new_image]
    if masks is not None: