[fbsync] Fix ternary operator to decide to store an image in Grayscale or RGB (#3553)

Summary: Co-authored-by: Vasilis Vryniotis <[email protected]>

Reviewed By: fmassa

Differential Revision: D27127992

fbshipit-source-id: 76ef726f854788ce71e785a838637e521d1f422c

Author: vincentqb

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,