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

◆ __init__()

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

◆ run_model_on_image()

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.

◆ run_processed_image()

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

◆ hypes

segmentation.kittiseg.KittiSegmenter.hypes

Definition at line 75 of file kittiseg.py.

◆ output_image_uncropped

segmentation.kittiseg.KittiSegmenter.output_image_uncropped

Definition at line 124 of file kittiseg.py.

◆ tf_image_pl

segmentation.kittiseg.KittiSegmenter.tf_image_pl

Definition at line 91 of file kittiseg.py.

◆ tf_prediction

segmentation.kittiseg.KittiSegmenter.tf_prediction

Definition at line 96 of file kittiseg.py.

◆ tf_session

segmentation.kittiseg.KittiSegmenter.tf_session

Definition at line 115 of file kittiseg.py.

◆ tools

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 Feb 28 2022 22:04:10