component_memory.h

Go to the documentation of this file.

// Copyright (c) 2024, DaoAI Robotics. All rights reserved.
// Version: 2025.3.0
 
#pragma once
#include <filesystem>
 
namespace DaoAI
{
 namespace DeepLearning
 {
 namespace Vision
 {
 class UnsupervisedDefectSegmentationImpl;
 }
 
 class ComponentMemoryImpl;
 
 class ComponentMemory
 {
 public:
 ComponentMemory();
 
 ComponentMemory(std::shared_ptr<ComponentMemoryImpl> impl);
 
 DAOAI_API ComponentMemory(const std::filesystem::path& file_path);
 
 DAOAI_API void setImageThreshold(const float& value);
 
 DAOAI_API float getImageThreshold() const;
 
 DAOAI_API void setPixelThreshold(const float& value);
 
 DAOAI_API float getPixelThreshold() const;
 
 DAOAI_API void save(const std::filesystem::path& save_path) const;
 
 DAOAI_API void load(const std::filesystem::path& file_path);
 
 private:
 std::shared_ptr<ComponentMemoryImpl> model_instance_ = nullptr;
 
 friend class Vision::UnsupervisedDefectSegmentationImpl;
 };
 }
}