reusex::vision::Dataloader Class Reference

#include <Dataloader.hpp>

Classes
class	Iterator

Public Types
using	Pair = IDataset::Pair
using	Batch = std::vector<Pair>
using	BatchView = std::span<Pair>

Public Member Functions
	Dataloader (IDataset &dataset, size_t batch_size, bool shuffle=false, size_t num_workers=4, size_t prefetch_batches=2, std::optional< uint32_t > seed=default_seed)
	~Dataloader ()
Iterator	begin ()
Iterator	end ()
size_t	size () const
void	set_num_workers (size_t num_workers)
void	set_prefetch_batches (size_t prefetch_batches)
size_t	get_num_workers () const
size_t	get_prefetch_batches () const

Static Public Member Functions
static std::vector< size_t >	shuffled_indices (size_t count, std::optional< uint32_t > seed)

Static Public Attributes
static constexpr uint32_t	default_seed = 42

Detailed Description

Definition at line 29 of file Dataloader.hpp.

Member Typedef Documentation

Batch

using reusex::vision::Dataloader::Batch = std::vector<Pair>

Definition at line 32 of file Dataloader.hpp.

BatchView

using reusex::vision::Dataloader::BatchView = std::span<Pair>

Definition at line 33 of file Dataloader.hpp.

Pair

using reusex::vision::Dataloader::Pair = IDataset::Pair

Definition at line 31 of file Dataloader.hpp.

Constructor & Destructor Documentation

Dataloader()

reusex::vision::Dataloader::Dataloader	(	IDataset &	dataset,
		size_t	batch_size,
		bool	shuffle = false,
		size_t	num_workers = 4,
		size_t	prefetch_batches = 2,
		std::optional< uint32_t >	seed = default_seed )

References default_seed.

Referenced by reusex::vision::Dataloader::Iterator::Iterator(), and reusex::vision::Dataloader::Iterator::operator!=().

~Dataloader()

reusex::vision::Dataloader::~Dataloader

(

)

Member Function Documentation

begin()

Iterator reusex::vision::Dataloader::begin

(

)

end()

Iterator reusex::vision::Dataloader::end

(

)

get_num_workers()

size_t reusex::vision::Dataloader::get_num_workers

(

)

const

get_prefetch_batches()

size_t reusex::vision::Dataloader::get_prefetch_batches

(

)

const

set_num_workers()

void reusex::vision::Dataloader::set_num_workers

(

size_t

num_workers

)

set_prefetch_batches()

void reusex::vision::Dataloader::set_prefetch_batches

(

size_t

prefetch_batches

)

shuffled_indices()

std::vector< size_t > reusex::vision::Dataloader::shuffled_indices ( size_t count, std::optional< uint32_t > seed )

static

size()

size_t reusex::vision::Dataloader::size

(

)

const

Member Data Documentation

default_seed

uint32_t reusex::vision::Dataloader::default_seed = 42

staticconstexpr

Definition at line 40 of file Dataloader.hpp.

Referenced by Dataloader().

---

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

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