c_cpp 用于在图像上绘制面部标记的示例

Posted 2021-05-14

int main() {
    std::string landsFile = "land_list.txt";
    std::string imageFile = "image.jpg";
    // Define the rect
    // cv::Rect r(130, 154, 208, 208);

    // Load landmarsk from file
    std::vector<cv::Point2f> lands;
    std::string line;
    std::ifstream is(landsFile.c_str());
    while (std::getline(is, line))
    {
        std::istringstream iss(line);
        float x, y, z;
        if (!(iss >> x >> y >> z)) { break; } // error

        lands.push_back(cv::Point2f(x, y));
    }

    float scale = 1.7;  // (r.width + r.height) / 195;
    float scalex = scale;
    float scaley = scale;
    float offsetx = 14;
    float offsety = 14;
    // offsety = r.height * 0.12;
    size_t n = lands.size();
    for (size_t i = 0; i < n; ++i) {
        lands[i].x = (lands[i].x * scalex) + offsetx;
        lands[i].y = (lands[i].y * scaley) + offsety;
    }

    // Load image
    cv::Mat img = imread(imageFile);
    // cv::rectangle(img, r, cv::Scalar(255, 255, 255));

    // cv::resize(img, img, cv::Size(250, 250));
    // img = img(r).clone();

    // Draw lands on image
    int point_size = 3;
    for (size_t i = 0; i < n; ++i) {
        cv::circle(img, lands[i], point_size + 1, cv::Scalar(0, 0, 0), -1);
        cv::circle(img, lands[i], point_size, cv::Scalar(0, 255, 0), -1);
    }



    cv::imshow("Img", img);
    int res = cv::waitKey(0);

    if (static_cast<char>(res) == 's') {
        cv::imwrite("img.png", img);
    }

    return 0;
}