LibTorch dataset for RTABMap databases. More...

#include <Dataset.hpp>

Inheritance diagram for reusex::vision::libtorch::LibTorchDataset:

Collaboration diagram for reusex::vision::libtorch::LibTorchDataset:

Public Member Functions
IDataset::Pair	get (const std::size_t index) const override
	Get a data sample at the given index.
bool	save (const std::span< Pair > &data) override
	Save processed results back to the database.
	IDataset (std::shared_ptr< ProjectDB > database)
	IDataset (std::filesystem::path dbPath)
Public Member Functions inherited from reusex::vision::IDataset
	IDataset (std::shared_ptr< ProjectDB > database)
	IDataset (std::filesystem::path dbPath)
virtual	~IDataset ()=default
size_t	size () const

Additional Inherited Members
Public Types inherited from reusex::vision::IDataset
using	Pair = std::pair<std::unique_ptr<IData>, size_t>
Protected Member Functions inherited from reusex::vision::IDataset
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

LibTorch dataset for RTABMap databases.

Implements the IDataset interface for use with the backend-agnostic annotation pipeline. Loads images, applies letterbox preprocessing, and stores results as LibTorchData objects.

Definition at line 18 of file Dataset.hpp.

Member Function Documentation

◆ get()

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

overridevirtual

Get a data sample at the given index.

Parameters

index Index in the dataset (0 to size()-1)

Returns: Pair containing LibTorchData with letterboxed image and the index

Implements reusex::vision::IDataset.

◆ IDataset() [1/2]

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

explicit

◆ IDataset() [2/2]

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

explicit

◆ save()

bool reusex::vision::libtorch::LibTorchDataset::save ( const std::span< Pair > & data )

overridevirtual

Save processed results back to the database.