From dc2e5f15050889a228d56644fd911e186d7890b9 Mon Sep 17 00:00:00 2001 From: Daniel Keysers Date: Tue, 1 Oct 2024 04:52:57 -0700 Subject: [PATCH] PaliGemma: fix image loading. Use uint8_t to make sure values are not interpreted as signed char. PiperOrigin-RevId: 680965115 --- paligemma/image.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/paligemma/image.cc b/paligemma/image.cc index 716436b..309296c 100644 --- a/paligemma/image.cc +++ b/paligemma/image.cc @@ -15,6 +15,8 @@ #include "paligemma/image.h" +#include + #include #include #include @@ -97,8 +99,8 @@ bool Image::ReadPPM(const std::string& filename) { height_ = height; int data_size = width * height * 3; data_.resize(data_size); - std::vector data_bytes(data_size); - file.read(data_bytes.data(), data_size); + std::vector data_bytes(data_size); + file.read(reinterpret_cast(data_bytes.data()), data_size); if (file.gcount() != data_size) { std::cerr << "Failed to read " << data_size << " bytes\n"; return false;