jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset Class Reference

Inheritance diagram for jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset:

Public Member Functions
def	__init__ (self, root_dir, split, aug=False)
def	__len__ (self)
def	get_example (self, i)
def	rotate_depth_image (self, in_img, angle, flags=cv2.INTER_LINEAR)
def	rotate_image (self, in_img, angle, flags=cv2.INTER_LINEAR)

Public Attributes
	aug
	root_dir
	split

Static Public Attributes
	class_names
float	max_value = 5.0
	mean_bgr = np.array([104.00698793, 116.66876762, 122.67891434])
float	min_value = 0.5
	write

Private Member Functions
def	_get_example (self, files_dir, idx)

Private Attributes
	_files_dirs
	_scenes

Static Private Attributes
	_files

Detailed Description

Definition at line 13 of file depth_prediction.py.

Constructor & Destructor Documentation

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.__init__	(		self,
			root_dir,
			split,
			aug = False
	)

Definition at line 32 of file depth_prediction.py.

Member Function Documentation

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.__len__

(

self

)

Definition at line 52 of file depth_prediction.py.

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset._get_example (   self,   files_dir,   idx  )

private

Definition at line 58 of file depth_prediction.py.

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.get_example	(		self,
			i
	)

Definition at line 55 of file depth_prediction.py.

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.rotate_depth_image	(		self,
			in_img,
			angle,
			flags = cv2.INTER_LINEAR
	)

Definition at line 172 of file depth_prediction.py.

def jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.rotate_image	(		self,
			in_img,
			angle,
			flags = cv2.INTER_LINEAR
	)

Definition at line 163 of file depth_prediction.py.

Member Data Documentation

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset._files

staticprivate

Initial value:

= set([
        'depth.npz',
        'depth_gt.npz',
        'image.png',
        'label.png'
    ])

Definition at line 21 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset._files_dirs

private

Definition at line 38 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset._scenes

private

Definition at line 39 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.aug

Definition at line 36 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.class_names

static

Initial value:

= np.array([
        '_background_',
        'mirror',
    ], dtype=np.str)

Definition at line 15 of file depth_prediction.py.

float jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.max_value = 5.0

static

Definition at line 30 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.mean_bgr = np.array([104.00698793, 116.66876762, 122.67891434])

static

Definition at line 28 of file depth_prediction.py.

float jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.min_value = 0.5

static

Definition at line 29 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.root_dir

Definition at line 33 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.split

Definition at line 35 of file depth_prediction.py.

jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset.write

static

Definition at line 19 of file depth_prediction.py.

---

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

• depth_prediction.py