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.
77 lines
2.1 KiB
77 lines
2.1 KiB
2 months ago
|
// License: Apache 2.0. See LICENSE file in root directory.
|
||
|
|
// Copyright(c) 2019 Intel Corporation. All Rights Reserved.
|
||
|
|
|
||
|
|
#include <rs-vino/base-detection.h>
|
||
|
|
#include <rsutils/easylogging/easyloggingpp.h>
|
||
|
|
|
||
|
|
|
||
|
|
using namespace InferenceEngine;
|
||
|
|
|
||
|
|
|
||
|
|
namespace openvino_helpers
|
||
|
|
{
|
||
|
|
base_detection::base_detection(
|
||
|
|
std::string topoName,
|
||
|
|
const std::string &pathToModel,
|
||
|
|
int maxBatch, bool isBatchDynamic, bool isAsync,
|
||
|
|
bool doRawOutputMessages
|
||
|
|
)
|
||
|
|
: topoName( topoName ), pathToModel( pathToModel )
|
||
|
|
, maxBatch( maxBatch ), isBatchDynamic( isBatchDynamic ), isAsync( isAsync )
|
||
|
|
, enablingChecked( false ), _enabled( false ), doRawOutputMessages( doRawOutputMessages )
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void base_detection::submit_request()
|
||
|
|
{
|
||
|
|
if( !enabled() || !_request )
|
||
|
|
return;
|
||
|
|
if( isAsync )
|
||
|
|
_request->StartAsync();
|
||
|
|
else
|
||
|
|
_request->Infer();
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void base_detection::wait()
|
||
|
|
{
|
||
|
|
if( !enabled() || !_request || !isAsync )
|
||
|
|
return;
|
||
|
|
_request->Wait( IInferRequest::WaitMode::RESULT_READY );
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
bool base_detection::enabled() const
|
||
|
|
{
|
||
|
|
if( !enablingChecked )
|
||
|
|
{
|
||
|
|
_enabled = !pathToModel.empty();
|
||
|
|
if( !_enabled )
|
||
|
|
LOG(INFO) << topoName + " DISABLED";
|
||
|
|
enablingChecked = true;
|
||
|
|
}
|
||
|
|
return _enabled;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void base_detection::load_into( InferenceEngine::Core & ie, const std::string & deviceName )
|
||
|
|
{
|
||
|
|
if( !enabled() )
|
||
|
|
return;
|
||
|
|
|
||
|
|
std::map<std::string, std::string> config = { };
|
||
|
|
if( isBatchDynamic )
|
||
|
|
{
|
||
|
|
bool isPossibleDynBatch =
|
||
|
|
deviceName.find( "CPU" ) != std::string::npos ||
|
||
|
|
deviceName.find( "GPU" ) != std::string::npos;
|
||
|
|
if( isPossibleDynBatch )
|
||
|
|
config[PluginConfigParams::KEY_DYN_BATCH_ENABLED] = PluginConfigParams::YES;
|
||
|
|
}
|
||
|
|
|
||
|
|
//LOG(INFO) << "Loading " << topoName << " model to the " << deviceName << " device";
|
||
|
|
net = ie.LoadNetwork( read_network(), deviceName, config );
|
||
|
|
}
|
||
|
|
}
|