基于与Google Gemini交互获取的Opencv在图片上写汉字的实现

sudo apt-get install libfreetype6-dev
sudo apt-get install fonts-wqy-zenhei

CMakeLists.txt

cmake_minimum_required(VERSION 3.10) # Or a more recent version

project(OpenCVChineseText)

set(CMAKE_CXX_STANDARD 17) # Or your preferred C++ standard
set(CMAKE_CXX_STANDARD_REQUIRED ON)

# --- Find OpenCV ---
find_package(OpenCV REQUIRED)
if(OpenCV_FOUND)
    message(STATUS "Found OpenCV: ${OpenCV_DIR}")
    include_directories(${OpenCV_INCLUDE_DIRS})
else()
    message(FATAL_ERROR "OpenCV not found. Please set OpenCV_DIR or ensure it's in your PATH.")
endif()

# --- Find FreeType ---
find_package(Freetype REQUIRED)
if(Freetype_FOUND)
    message(STATUS "Found FreeType: ${FREETYPE_DIR}")
    include_directories(${FREETYPE_INCLUDE_DIRS})
else()
    message(FATAL_ERROR "FreeType not found. Please ensure it's installed.")
endif()

# --- Add Executable ---
add_executable(opencv_chinese_text main.cpp) # Replace main.cpp with your source file(s)

# --- Link Libraries ---
target_link_libraries(opencv_chinese_text
    ${OpenCV_LIBS}
    ${FREETYPE_LIBRARIES}
)

# --- Optional: Install target ---
install(TARGETS opencv_chinese_text DESTINATION bin)
install(DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/fonts DESTINATION share/opencv_chinese_text) # If you have a fonts folder

main.cpp

#include <iostream>
#include <string>
#include <vector>
#include <ft2build.h>
#include FT_FREETYPE_H
#include <opencv2/opencv.hpp>
#include <locale>
#include <codecvt>

int main()
{
    cv::Mat image = cv::Mat::zeros(200, 500, CV_8UC3);
    std::string chineseText = "你好，世界";
    int fontSize = 48;
    int x = 50, y = 100;
    cv::Scalar textColor(255, 255, 255); // White color for text

    FT_Library library;
    if (FT_Init_FreeType(&library))
    {
        std::cerr << "Error initializing FreeType library" << std::endl;
        return -1;
    }

    FT_Face face;
    if (FT_New_Face(library, "/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc", 0, &face))
    {
        std::cerr << "Error loading font file" << std::endl;
        FT_Done_FreeType(library);
        return -1;
    }

    FT_Set_Pixel_Sizes(face, 0, fontSize);

    std::wstring_convert<std::codecvt_utf8<wchar_t>> converter;
    std::wstring wide_text = converter.from_bytes(chineseText);

    for (wchar_t wc : wide_text)
    {
        FT_UInt glyph_index = FT_Get_Char_Index(face, wc);
        if (FT_Load_Glyph(face, glyph_index, FT_LOAD_RENDER))
        {
            std::cerr << "Error loading glyph" << std::endl;
            continue;
        }

        FT_Bitmap &bitmap = face->glyph->bitmap;

        for (int i = 0; i < bitmap.rows; ++i)
        {
            for (int j = 0; j < bitmap.width; ++j)
            {
                unsigned char pixel_value = bitmap.buffer[i * bitmap.pitch + j];
                if (pixel_value > 0)
                {
                    cv::Point draw_point(x + face->glyph->bitmap_left + j, y - face->glyph->bitmap_top + i);
                    if (draw_point.x >= 0 && draw_point.x < image.cols && draw_point.y >= 0 && draw_point.y < image.rows)
                    {
                        image.at<cv::Vec3b>(draw_point)[0] = cv::saturate_cast<uchar>(textColor[0] * (pixel_value / 255.0));
                        image.at<cv::Vec3b>(draw_point)[1] = cv::saturate_cast<uchar>(textColor[1] * (pixel_value / 255.0));
                        image.at<cv::Vec3b>(draw_point)[2] = cv::saturate_cast<uchar>(textColor[2] * (pixel_value / 255.0));
                    }
                }
            }
        }

        x += face->glyph->advance.x >> 6;
    }

    FT_Done_Face(face);
    FT_Done_FreeType(library);

    cv::putText(image, "This is English", cv::Point(50, 50), cv::FONT_HERSHEY_SIMPLEX, 1.0, cv::Scalar(255, 255, 255));
    cv::imshow("Image with Text", image);
    cv::waitKey(0);
    cv::destroyAllWindows();

    return 0;
}