Backend.hpp

Go to the documentation of this file.

#pragma once
#include <ReUseX/vision/IMLBackend.hpp>
#include <ReUseX/vision/tensor_rt/Dataset.hpp>
#include <ReUseX/vision/tensor_rt/Sam3.hpp>
 
namespace ReUseX::vision::tensor_rt {
/* TensorRTBackend is a concrete implementation of the IMLBackend interface for
 * TensorRT. It provides methods to create TensorRT-based models and datasets.
 */
class TensorRTBackend : public IMLBackend {
    public:
  /* Constructor for TensorRTBackend. Initializes any necessary resources for
   * the backend. In this case, it is a default constructor.
   */
  TensorRTBackend() = default;
 
  /* Destructor for TensorRTBackend. Cleans up any resources allocated by the
   * backend. In this case, it is a default destructor.
   * @param: type - The type of model to create (not used in this
   * implementation).
   * @param: modelPath - The file path to the model to be created (not used in
   * this implementation).
   * @return: A unique pointer to an IModel instance representing the created
   * model.
   */
  std::unique_ptr<IModel>
  createModel(const Model type,
              const std::filesystem::path &modelPath) override;
 
  /* Creates a dataset based on the provided dataset path. This method is
   * responsible for initializing and returning a dataset that can be used for
   * training or inference. The dataset is created based on the specified path,
   * which may contain the necessary data and configuration for the dataset.
   * @param: datasetPath - The file path to the dataset to be created.
   * @return: A unique pointer to an IDataset instance representing the created
   * dataset.
   */
  std::unique_ptr<IDataset>
  createDataset(const std::filesystem::path &datasetPath) override;
};
} // namespace ReUseX::vision::tensor_rt