Model.h

Go to the documentation of this file.

/*
 * Copyright (c) 2020, NVIDIA CORPORATION.  All rights reserved.
 *
 * NVIDIA CORPORATION and its licensors retain all intellectual property
 * and proprietary rights in and to this software, related documentation
 * and any modifications thereto.  Any use, reproduction, disclosure or
 * distribution of this software and related documentation without an express
 * license agreement from NVIDIA CORPORATION is strictly prohibited.
 */
 
#ifndef CVCORE_MODEL_H
#define CVCORE_MODEL_H
 
#include <vector>
 
#include "Image.h"
 
namespace cvcore {
 
struct ModelInputParams
{
    size_t maxBatchSize;      
    size_t inputLayerWidth;   
    size_t inputLayerHeight;  
    ImageType modelInputType; 
};
 
struct ModelInferenceParams
{
    std::string engineFilePath;            
    std::vector<std::string> inputLayers;  
    std::vector<std::string> outputLayers; 
};
 
} // namespace cvcore
 
#endif // CVCORE_MODEL_H