Class to initialize and run KittiSeg
Definition at line 65 of file kittiseg.py.
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.
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.
segmentation.kittiseg.KittiSegmenter.hypes |
segmentation.kittiseg.KittiSegmenter.output_image_uncropped |
segmentation.kittiseg.KittiSegmenter.tf_image_pl |
segmentation.kittiseg.KittiSegmenter.tf_prediction |
segmentation.kittiseg.KittiSegmenter.tf_session |
segmentation.kittiseg.KittiSegmenter.tools |
The documentation for this class was generated from the following file: