Public Member Functions | Public Attributes | Static Public Attributes | Private Member Functions | Private Attributes | Static Private Attributes | List of all members
jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset Class Reference
Inheritance diagram for jsk_recognition_utils.datasets.depth_prediction.DepthPredictionDataset:
Inheritance graph
[legend]

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:
1 = set([
2  'depth.npz',
3  'depth_gt.npz',
4  'image.png',
5  'label.png'
6  ])

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:
1 = np.array([
2  '_background_',
3  'mirror',
4  ], 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:


jsk_recognition_utils
Author(s):
autogenerated on Sat Mar 27 2021 03:02:11