Sam3Dataset.hpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: 2026 Povl Filip Sonne-Frederiksen
//
// SPDX-License-Identifier: GPL-3.0-or-later
 
#pragma once
#include "reusex/vision/IDataset.hpp"
#include "reusex/vision/onnx/Sam3Data.hpp"
 
namespace reusex::vision::onnx {

class ONNXSam3Dataset : public IDataset {
    public:
  using IDataset::IDataset;

  IDataset::Pair get(const std::size_t index) const override;

  bool save(const std::span<Pair> &data) override;
 
    private:
  bool class_map_saved_ = false;
};
 
} // namespace reusex::vision::onnx