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)
	~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

Detailed Description

Definition at line 28 of file Dataloader.hpp.

Member Typedef Documentation

Batch

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

Definition at line 31 of file Dataloader.hpp.

BatchView

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

Definition at line 32 of file Dataloader.hpp.

Pair

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

Definition at line 30 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 )

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

)

size()

size_t reusex::vision::Dataloader::size

(

)

const

---

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

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