LocalMapCNNDataset Class Reference

Dataset for CNN training. More...

Inheritance diagram for LocalMapCNNDataset:

Public Member Functions
	__init__ (self, str data_dir, str stage, bool use_comm_map, int output_dim, bool preload=True)
	len (self)
	get (self, idx)
	load_data (self)
	Load the data from the data directory.

Public Attributes
	stage = stage
	data_dir = data_dir
	output_dim = output_dim
	use_comm_map = use_comm_map
	maps
	dataset_size = self.maps.shape[0]
	targets
	targets_mean
tuple	targets_std

Detailed Description

Dataset for CNN training.

Definition at line 38 of file loaders.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
		str	data_dir,
		str	stage,
		bool	use_comm_map,
		int	output_dim,
		bool	preload = True )

Definition at line 40 of file loaders.py.

Member Function Documentation

get()

get	(		self,
			idx )

Definition at line 70 of file loaders.py.

len()

len

(

self

)

Definition at line 67 of file loaders.py.

load_data()

load_data

(

self

)

Load the data from the data directory.

Definition at line 79 of file loaders.py.

Member Data Documentation

data_dir

data_dir = data_dir

Definition at line 60 of file loaders.py.

dataset_size

dataset_size = self.maps.shape[0]

Definition at line 88 of file loaders.py.

maps

maps

Initial value:

=  DataLoaderUtils.load_maps(
                f"{self.data_dir}/{self.stage}", self.use_comm_map
                )

Definition at line 81 of file loaders.py.

output_dim

output_dim = output_dim

Definition at line 61 of file loaders.py.

stage

stage = stage

Definition at line 59 of file loaders.py.

targets

targets

Definition at line 91 of file loaders.py.

targets_mean

targets_mean

Definition at line 91 of file loaders.py.

targets_std

tuple targets_std

Initial value:

=  (
                DataLoaderUtils.load_actions(f"{self.data_dir}/{self.stage}")
                )

Definition at line 91 of file loaders.py.

use_comm_map

use_comm_map = use_comm_map

Definition at line 62 of file loaders.py.