mirror of
https://github.com/deepinsight/insightface.git
synced 2026-06-20 08:07:54 +00:00
Update inspireface to 1.2.0
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
#include <iostream>
|
||||
#include <inspirecv/inspirecv.h>
|
||||
#include "inspireface/initialization_module/launch.h"
|
||||
#include <inspireface/middleware/inspirecv_image_process.h>
|
||||
#include "inspireface/track_module/landmark/face_landmark_adapt.h"
|
||||
int main() {
|
||||
std::string expansion_path = "";
|
||||
INSPIRE_LAUNCH->Load("test_res/pack/Pikachu-t4");
|
||||
auto archive = INSPIRE_LAUNCH->getMArchive();
|
||||
|
||||
inspire::InspireModel lmkModel;
|
||||
auto ret = archive.LoadModel("landmark", lmkModel);
|
||||
if (ret != 0) {
|
||||
INSPIRE_LOGE("Load %s error: %d", "landmark", ret);
|
||||
return -1;
|
||||
}
|
||||
|
||||
inspire::FaceLandmarkAdapt lmk;
|
||||
lmk.loadData(lmkModel, lmkModel.modelType);
|
||||
|
||||
auto image = inspirecv::Image::Create("test_res/data/crop/crop.png");
|
||||
auto data = image.Resize(112, 112);
|
||||
auto lmk_out = lmk(data);
|
||||
std::vector<inspirecv::Point2i> landmarks_output(inspire::FaceLandmarkAdapt::NUM_OF_LANDMARK);
|
||||
for (int i = 0; i < inspire::FaceLandmarkAdapt::NUM_OF_LANDMARK; ++i) {
|
||||
float x = lmk_out[i * 2 + 0] * image.Width();
|
||||
float y = lmk_out[i * 2 + 1] * image.Height();
|
||||
landmarks_output[i] = inspirecv::Point<int>(x, y);
|
||||
}
|
||||
|
||||
for (int i = 0; i < landmarks_output.size(); ++i) {
|
||||
image.DrawCircle(landmarks_output[i], 5, {0, 0, 255});
|
||||
}
|
||||
image.Write("crop_lmk.png");
|
||||
}
|
||||
Reference in New Issue
Block a user