Skip to content

GitLab

Commit f858af81 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TF Object Detection Team
Browse files

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

PiperOrigin-RevId: 341573525
parent 0b7646e2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/object_detection/core/preprocessor.py
View file @ f858af81
...@@ -2937,7 +2937,7 @@ def resize_to_range(image, ...@@ -2937,7 +2937,7 @@ def resize_to_range(image,
for i in range(len(channels)) for i in range(len(channels))
], ],
axis=2) axis=2)
new_image.set_shape([max_dimension, max_dimension, 3]) new_image.set_shape([max_dimension, max_dimension, len(channels)])
result = [new_image] result = [new_image]
if masks is not None: if masks is not None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment