Public Member Functions | Public Attributes | List of all members
segmentation.kittiseg.KittiSegmenter Class Reference
Inheritance diagram for segmentation.kittiseg.KittiSegmenter:
Inheritance graph
[legend]

Public Member Functions

def __init__ (self, logdir, original_image_size, runCPU=False)
 
def run_model_on_image (self, image)
 
def run_processed_image (self, image)
 

Public Attributes

 hypes
 
 output_image_uncropped
 
 tf_image_pl
 
 tf_prediction
 
 tf_session
 
 tools
 

Detailed Description

Class to initialize and run KittiSeg

Definition at line 65 of file kittiseg.py.

Constructor & Destructor Documentation

def segmentation.kittiseg.KittiSegmenter.__init__ (   self,
  logdir,
  original_image_size,
  runCPU = False 
)
Initialize KittiSeg:
    Inputs: logdir (Directory to KittiSeg model),
    runCPU (Optional, if set to True TensorFlow runs on CPU instead of GPU

Definition at line 67 of file kittiseg.py.

Member Function Documentation

def segmentation.kittiseg.KittiSegmenter.run_model_on_image (   self,
  image 
)
A function that sets up and runs an image through KittiSeg
Input: Image to process
Output: way_prediction, time_tf

Definition at line 143 of file kittiseg.py.

def segmentation.kittiseg.KittiSegmenter.run_processed_image (   self,
  image 
)
A function that runs an image through KittiSeg
Input: Image to process
Output: tagged_image, time_tf

Definition at line 126 of file kittiseg.py.

Member Data Documentation

segmentation.kittiseg.KittiSegmenter.hypes

Definition at line 75 of file kittiseg.py.

segmentation.kittiseg.KittiSegmenter.output_image_uncropped

Definition at line 124 of file kittiseg.py.

segmentation.kittiseg.KittiSegmenter.tf_image_pl

Definition at line 91 of file kittiseg.py.

segmentation.kittiseg.KittiSegmenter.tf_prediction

Definition at line 96 of file kittiseg.py.

segmentation.kittiseg.KittiSegmenter.tf_session

Definition at line 115 of file kittiseg.py.

segmentation.kittiseg.KittiSegmenter.tools

Definition at line 123 of file kittiseg.py.


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


cnn_bridge
Author(s): Noam C. Golombek , Alexander Beringolts
autogenerated on Mon Jun 10 2019 12:53:26