msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl Class Reference

Inheritance diagram for msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl:

Public Member Functions
	__init__ (self, files_path, channel=0)
	Args:
	get_sample (self, entry)
	Args:
Public Member Functions inherited from msecnn_raulkviana.custom_dataset.CUDatasetBase
	__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
	channel
Public Attributes inherited from msecnn_raulkviana.custom_dataset.CUDatasetBase
	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. For stage 2

This version contains all of the information available from the CU data

Constructor & Destructor Documentation

__init__()

msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl.__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 from msecnn_raulkviana.custom_dataset.CUDatasetBase.

Member Function Documentation

get_sample()

msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl.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 from msecnn_raulkviana.custom_dataset.CUDatasetBase.

Member Data Documentation

channel

msecnn_raulkviana.custom_dataset.CUDatasetStg2Compl.channel

---

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

• src/msecnn_raulkviana/custom_dataset.py