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 (boost::program_options::options_description &options) 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 181 of file YoloInferenceTest.hpp.

Constructor & Destructor Documentation

YoloTestCaseProvider()

YoloTestCaseProvider ( TConstructModelCallable  constructModel )

inlineexplicit

Definition at line 185 of file YoloInferenceTest.hpp.

        : m_ConstructModel(constructModel)
    {
    }

Member Function Documentation

AddCommandLineOptions()

virtual void AddCommandLineOptions ( boost::program_options::options_description &  options )

inlineoverridevirtual

Reimplemented from IInferenceTestCaseProvider.

Definition at line 190 of file YoloInferenceTest.hpp.

    {
        namespace po = boost::program_options;

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

        Model::AddCommandLineOptions(options, m_ModelCommandLineOptions);
    }

GetTestCase()

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

inlineoverridevirtual

Implements IInferenceTestCaseProvider.

Definition at line 223 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 201 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