YoloTestCaseProvider< Model > Class Template Reference

#include <YoloInferenceTest.hpp>

Inheritance diagram for YoloTestCaseProvider< Model >:

Public Member Functions
template<typename TConstructModelCallable >
	YoloTestCaseProvider (TConstructModelCallable constructModel)
virtual void	AddCommandLineOptions (cxxopts::Options &options, std::vector< std::string > &required) override
virtual bool	ProcessCommandLineOptions (const InferenceTestOptions &commonOptions) override
virtual std::unique_ptr< IInferenceTestCase >	GetTestCase (unsigned int testCaseId) override
Public Member Functions inherited from IInferenceTestCaseProvider
virtual	~IInferenceTestCaseProvider ()
virtual bool	OnInferenceTestFinished ()

Detailed Description

template<typename Model>
class YoloTestCaseProvider< Model >

Definition at line 177 of file YoloInferenceTest.hpp.

Constructor & Destructor Documentation

YoloTestCaseProvider()

YoloTestCaseProvider ( TConstructModelCallable  constructModel )

inlineexplicit

Definition at line 181 of file YoloInferenceTest.hpp.

        : m_ConstructModel(constructModel)
    {
    }

Member Function Documentation

AddCommandLineOptions()

virtual void AddCommandLineOptions ( cxxopts::Options &  options, std::vector< std::string > &  required  )

inlineoverridevirtual

Reimplemented from IInferenceTestCaseProvider.

Definition at line 186 of file YoloInferenceTest.hpp.

    {
        options
            .allow_unrecognised_options()
            .add_options()
                ("d,data-dir", "Path to directory containing test data", cxxopts::value<std::string>(m_DataDir));

        Model::AddCommandLineOptions(options, m_ModelCommandLineOptions, required);
    }

GetTestCase()

virtual std::unique_ptr<IInferenceTestCase> GetTestCase ( unsigned int  testCaseId )

inlineoverridevirtual

Implements IInferenceTestCaseProvider.

Definition at line 218 of file YoloInferenceTest.hpp.

    {
        std::unique_ptr<YoloTestCaseData> testCaseData = m_Database->GetTestCaseData(testCaseId);
        if (!testCaseData)
        {
            return nullptr;
        }

        return std::make_unique<YoloTestCase<Model>>(*m_Model, testCaseId, *testCaseData);
    }

ProcessCommandLineOptions()

virtual bool ProcessCommandLineOptions ( const InferenceTestOptions &  commonOptions )

inlineoverridevirtual

Reimplemented from IInferenceTestCaseProvider.

Definition at line 196 of file YoloInferenceTest.hpp.

References armnn::test::ValidateDirectory().

    {
        if (!ValidateDirectory(m_DataDir))
        {
            return false;
        }

        m_Model = m_ConstructModel(commonOptions, m_ModelCommandLineOptions);
        if (!m_Model)
        {
            return false;
        }

        m_Database = std::make_unique<YoloDatabase>(m_DataDir.c_str());
        if (!m_Database)
        {
            return false;
        }

        return true;
    }

---

The documentation for this class was generated from the following file:

• tests/YoloInferenceTest.hpp