mirror of
https://github.com/deepinsight/insightface.git
synced 2026-04-09 10:30:36 +00:00
Update
This commit is contained in:
@@ -91,7 +91,7 @@ int main() {
|
||||
m_extract_ = std::make_shared<Extract>();
|
||||
InspireModel model;
|
||||
loader.LoadModel("feature", model);
|
||||
m_extract_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_extract_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
cv::Mat image = cv::imread(names[0]);
|
||||
// cv::Mat rgb;
|
||||
|
||||
@@ -31,7 +31,7 @@ int main() {
|
||||
|
||||
std::shared_ptr<FaceDetect> m_face_detector_;
|
||||
m_face_detector_ = std::make_shared<FaceDetect>(320);
|
||||
m_face_detector_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_face_detector_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
// Load a image
|
||||
cv::Mat image = cv::imread("test_res/images/face_sample.png");
|
||||
|
||||
@@ -35,7 +35,7 @@ void rec_function() {
|
||||
m_extract_ = std::make_shared<Extract>();
|
||||
InspireModel model;
|
||||
loader->LoadModel("feature", model);
|
||||
m_extract_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_extract_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
loader.reset();
|
||||
|
||||
|
||||
@@ -35,7 +35,7 @@ void test_rnet() {
|
||||
InspireModel model;
|
||||
loader.LoadModel("refine_net", model);
|
||||
m_rnet_ = std::make_shared<RNet>();
|
||||
m_rnet_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_rnet_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
{
|
||||
// Load a image
|
||||
@@ -78,7 +78,7 @@ void test_mask() {
|
||||
m_mask_predict_ = std::make_shared<MaskPredict>();
|
||||
InspireModel model;
|
||||
loader.LoadModel("mask_detect", model);
|
||||
m_mask_predict_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_mask_predict_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
{
|
||||
// Load a image
|
||||
@@ -120,7 +120,7 @@ void test_quality() {
|
||||
m_face_quality_ = std::make_shared<FacePoseQuality>();
|
||||
InspireModel model;
|
||||
loader.LoadModel("pose_quality", model);
|
||||
m_face_quality_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_face_quality_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
{
|
||||
std::vector<std::string> names = {
|
||||
@@ -166,7 +166,7 @@ void test_landmark_mnn() {
|
||||
m_landmark_predictor_ = std::make_shared<FaceLandmark>(112);
|
||||
InspireModel model;
|
||||
loader.LoadModel("landmark", model);
|
||||
m_landmark_predictor_->loadData(model);
|
||||
m_landmark_predictor_->LoadData(model);
|
||||
|
||||
cv::Mat image = cv::imread("test_res/images/test_data/crop.png");
|
||||
cv::resize(image, image, cv::Size(112, 112));
|
||||
@@ -206,7 +206,7 @@ void test_landmark() {
|
||||
m_landmark_predictor_ = std::make_shared<FaceLandmark>(112);
|
||||
InspireModel model;
|
||||
loader.LoadModel("landmark", model);
|
||||
m_landmark_predictor_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_landmark_predictor_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
cv::Mat image = cv::imread("test_res/images/test_data/0.jpg");
|
||||
cv::resize(image, image, cv::Size(112, 112));
|
||||
@@ -248,7 +248,7 @@ void test_liveness() {
|
||||
InspireModel model;
|
||||
loader.LoadModel("rgb_anti_spoofing", model);
|
||||
m_rgb_anti_spoofing_ = std::make_shared<RBGAntiSpoofing>(80, true);
|
||||
m_rgb_anti_spoofing_->loadData(model, InferenceWrapper::INFER_RKNN);
|
||||
m_rgb_anti_spoofing_->LoadData(model, InferenceWrapper::INFER_RKNN);
|
||||
|
||||
std::vector<std::string> names = {
|
||||
"test_res/images/test_data/real.jpg", "test_res/images/test_data/fake.jpg", "test_res/images/test_data/live.jpg",
|
||||
|
||||
Reference in New Issue
Block a user