reusex::vision::IDataset Class Reference

#include <IDataset.hpp>

Inheritance diagram for reusex::vision::IDataset:

Public Types
using	Pair = std::pair<std::unique_ptr<IData>, size_t>

Public Member Functions
	IDataset (std::shared_ptr< ProjectDB > database)
	IDataset (std::filesystem::path dbPath)
virtual	~IDataset ()=default
size_t	size () const
size_t	filter_annotated ()
	Remove already-annotated frames from the dataset.
virtual Pair	get (const std::size_t index) const =0
virtual bool	save (const std::span< Pair > &data)=0

Protected Member Functions
cv::Mat	image (const std::size_t index) const
bool	save_image (const std::size_t index, const cv::Mat &image)
std::shared_ptr< ProjectDB >	database () const

Detailed Description

Definition at line 35 of file IDataset.hpp.

Member Typedef Documentation

Pair

using reusex::vision::IDataset::Pair = std::pair<std::unique_ptr<IData>, size_t>

Definition at line 43 of file IDataset.hpp.

Constructor & Destructor Documentation

IDataset() [1/2]

reusex::vision::IDataset::IDataset ( std::shared_ptr< ProjectDB > database )

explicit

References database().

IDataset() [2/2]

reusex::vision::IDataset::IDataset ( std::filesystem::path dbPath )

explicit

~IDataset()

virtual reusex::vision::IDataset::~IDataset ( )

virtualdefault

Member Function Documentation

database()

std::shared_ptr< ProjectDB > reusex::vision::IDataset::database ( ) const

protected

Referenced by IDataset().

filter_annotated()

size_t reusex::vision::IDataset::filter_annotated

(

)

Remove already-annotated frames from the dataset.

Queries the database for frames that already have segmentation images and removes them from this dataset's ID list so they won't be processed.

Returns

Number of frames removed

get()

virtual Pair reusex::vision::IDataset::get ( const std::size_t index ) const

pure virtual

Implemented in reusex::vision::libtorch::LibTorchDataset, reusex::vision::onnx::ONNXSam3Dataset, and reusex::vision::tensor_rt::TensorRTDataset.

image()

cv::Mat reusex::vision::IDataset::image ( const std::size_t index ) const

protected

Referenced by save_image().

save()

virtual bool reusex::vision::IDataset::save ( const std::span< Pair > & data )

pure virtual

Implemented in reusex::vision::libtorch::LibTorchDataset, reusex::vision::onnx::ONNXSam3Dataset, and reusex::vision::tensor_rt::TensorRTDataset.

save_image()

bool reusex::vision::IDataset::save_image ( const std::size_t index, const cv::Mat & image )

protected

References image().

size()

size_t reusex::vision::IDataset::size

(

)

const

---

The documentation for this class was generated from the following file:

• libs/reusex/include/vision/IDataset.hpp