You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
977 B
42 lines
977 B
// License: Apache 2.0. See LICENSE file in root directory.
|
|
// Copyright(c) 2019 Intel Corporation. All Rights Reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "base-detection.h"
|
|
|
|
|
|
namespace openvino_helpers
|
|
{
|
|
/*
|
|
Detects age and gender for a give face
|
|
*/
|
|
class age_gender_detection : public base_detection
|
|
{
|
|
public:
|
|
struct Result
|
|
{
|
|
float age;
|
|
float maleProb;
|
|
};
|
|
|
|
private:
|
|
std::string input;
|
|
std::string outputAge;
|
|
std::string outputGender;
|
|
int _n_enqued_frames;
|
|
|
|
public:
|
|
age_gender_detection( const std::string &pathToModel,
|
|
bool isAsync = true,
|
|
int maxBatch = 1, bool isBatchDynamic = false,
|
|
bool doRawOutputMessages = false );
|
|
|
|
InferenceEngine::CNNNetwork read_network() override;
|
|
void submit_request() override;
|
|
|
|
void enqueue( const cv::Mat &face );
|
|
Result operator[] ( int idx ) const;
|
|
};
|
|
}
|