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
	data_dir
	output_dim
	use_comm_map
	maps
	dataset_size
	targets
	targets_mean
	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

Definition at line 60 of file loaders.py.

dataset_size

dataset_size

Definition at line 88 of file loaders.py.

maps

maps

Definition at line 81 of file loaders.py.

output_dim

output_dim

Definition at line 61 of file loaders.py.

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

targets_std

Definition at line 91 of file loaders.py.

use_comm_map

use_comm_map

Definition at line 62 of file loaders.py.