From 1bb0498eb011ac568380a5fcc05f64136ae58d6f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Edgar=20Andr=C3=A9s=20Margffoy=20Tuay?= Date: Thu, 11 Mar 2021 17:44:57 -0500 Subject: [PATCH] Fix ternary operator to decide to store an image in Grayscale or RGB --- torchvision/csrc/io/image/cpu/encode_png.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/csrc/io/image/cpu/encode_png.cpp b/torchvision/csrc/io/image/cpu/encode_png.cpp index 3cf5de6955b..d28bad95890 100644 --- a/torchvision/csrc/io/image/cpu/encode_png.cpp +++ b/torchvision/csrc/io/image/cpu/encode_png.cpp @@ -128,7 +128,7 @@ torch::Tensor encode_png(const torch::Tensor& data, int64_t compression_level) { png_set_write_fn(png_write, &buf_info, torch_png_write_data, NULL); // Set output image information - auto color_type = PNG_COLOR_TYPE_GRAY ? channels == 1 : PNG_COLOR_TYPE_RGB; + auto color_type = channels == 1 ? PNG_COLOR_TYPE_GRAY : PNG_COLOR_TYPE_RGB; png_set_IHDR( png_write, info_ptr,