msecnn_raulkviana.custom_dataset.CUDatasetBase Class Reference

Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again. More...

Inheritance diagram for msecnn_raulkviana.custom_dataset.CUDatasetBase:

Public Member Functions
	__init__ (self, files_path, channel=0)
	Args:
	__len__ (self)
	get_sample (self, entry)
	Args:
	obtain_files_sizes (self, files)
	Args:
	select_entry (self, idx)
	Args:
	__getitem__ (self, idx)

Public Attributes
	files_path
	files
	lst_entries_nums
	total_num_entries
	channel
	index_lims
	data_files

Detailed Description

Dataset stage oriented with capability of loading different files and it's supposed to be used with the function dataset_utils.change_labels_function_again.

Works for stage 2 and 3

Constructor & Destructor Documentation

__init__()

msecnn_raulkviana.custom_dataset.CUDatasetBase.__init__	(		self,
			files_path,
			channel = 0
	)

Args:

Parameters

files_path	(string): Path to the files with annotations.
channel	Channel to get for the dataset (0:Luma, 1:Chroma)

Reimplemented in msecnn_raulkviana.custom_dataset.CUDatasetStg5Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg6Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg4Compl, and msecnn_raulkviana.custom_dataset.CUDatasetStg3Compl.

Member Function Documentation

__getitem__()

msecnn_raulkviana.custom_dataset.CUDatasetBase.__getitem__	(		self,
			idx
	)

__len__()

msecnn_raulkviana.custom_dataset.CUDatasetBase.__len__

(

self

)

get_sample()

msecnn_raulkviana.custom_dataset.CUDatasetBase.get_sample	(		self,
			entry
	)

Args:

Parameters

entry

(int): An instance from the labels.

Returns

out: lst - CTU | RD_for_specific_stage | cu_left_of_stg_1 | cu_top_of_stg_1 | cu_left_for_specific_stage | cu_top_for_specific_stage | split_for_specific_stage

Reimplemented in msecnn_raulkviana.custom_dataset.CUDatasetStg5Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg6Compl, msecnn_raulkviana.custom_dataset.CUDatasetStg4Compl, and msecnn_raulkviana.custom_dataset.CUDatasetStg3Compl.

obtain_files_sizes()

msecnn_raulkviana.custom_dataset.CUDatasetBase.obtain_files_sizes	(		self,
			files
	)

Args:

Parameters

files

(list): List containing the names of files with CUs info

select_entry()

msecnn_raulkviana.custom_dataset.CUDatasetBase.select_entry	(		self,
			idx
	)

Args:

Parameters

idx	(int): Index with the position to search for a specific entry

Member Data Documentation

channel

msecnn_raulkviana.custom_dataset.CUDatasetBase.channel

data_files

msecnn_raulkviana.custom_dataset.CUDatasetBase.data_files

files

msecnn_raulkviana.custom_dataset.CUDatasetBase.files

files_path

msecnn_raulkviana.custom_dataset.CUDatasetBase.files_path

index_lims

msecnn_raulkviana.custom_dataset.CUDatasetBase.index_lims

lst_entries_nums

msecnn_raulkviana.custom_dataset.CUDatasetBase.lst_entries_nums

total_num_entries

msecnn_raulkviana.custom_dataset.CUDatasetBase.total_num_entries

---

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

• src/msecnn_raulkviana/custom_dataset.py