From 2272d7c5a5490a5b2405dbac75eb65a300eae041 Mon Sep 17 00:00:00 2001 From: cyy Date: Wed, 23 Jun 2021 21:01:43 +0800 Subject: [PATCH] fix 3613 --- torchvision/csrc/io/image/cpu/decode_png.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/torchvision/csrc/io/image/cpu/decode_png.cpp b/torchvision/csrc/io/image/cpu/decode_png.cpp index 315ed611614..b40fd951d5b 100644 --- a/torchvision/csrc/io/image/cpu/decode_png.cpp +++ b/torchvision/csrc/io/image/cpu/decode_png.cpp @@ -71,6 +71,11 @@ torch::Tensor decode_png(const torch::Tensor& data, ImageReadMode mode) { TORCH_CHECK(retval == 1, "Could read image metadata from content.") } + if (bit_depth > 8) { + png_destroy_read_struct(&png_ptr, &info_ptr, nullptr); + TORCH_CHECK(false, "At most 8-bit PNG images are supported currently.") + } + int channels = png_get_channels(png_ptr, info_ptr); if (color_type == PNG_COLOR_TYPE_GRAY && bit_depth < 8)