m5/StackFlow
0
0
Fork 0
mirror of https://github.com/m5stack/StackFlow.git synced 2026-05-20 11:32:11 -07:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of github.com:m5stack/StackFlow

Browse Source
This commit is contained in:
dianjixz
2024-11-19 11:55:07 +08:00
parent 4a2394ca70 466b7707bf
commit bbfbf1f4ca
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/llm_framework/main_vlm/src/main.cpp
+6
View File
@@ -198,6 +198,12 @@ class llm_task {
std::string out = lLaMa_->Run(prompt_data_);
if (out_callback_) out_callback_(out, true);
} else {
if (image_data_.size() > 2 && image_data_[0] == 0xFF && image_data_[1] == 0xD8 &&
image_data_[image_data_.size() - 2] == 0xFF && image_data_[image_data_.size() - 1] == 0xD9) {
} else {
SLOGE("Invalid JPEG data.");
return;
}
cv::Mat src = cv::imdecode(image_data_, cv::IMREAD_COLOR);
image_data_.clear();
lLaMa_->Encode(src, img_embed);