segmentation.deeplab_segmenter.DeepLabSegmenter Class Reference

Inheritance diagram for segmentation.deeplab_segmenter.DeepLabSegmenter:

Public Member Functions
def	__init__ (self, model_dir, original_image_size, tensor_io, runCPU, gpu_percent=1)
def	run_model_on_image (self, image)
def	run_processed_image (self, image)

Public Attributes
	graph
	hypes
	input_image_size
	input_tensor
	output_image_uncropped
	output_tensor
	sess
	tools

Detailed Description

Class to load deeplab model and run inference.

Definition at line 55 of file deeplab_segmenter.py.

Constructor & Destructor Documentation

def segmentation.deeplab_segmenter.DeepLabSegmenter.__init__	(		self,
			model_dir,
			original_image_size,
			tensor_io,
			runCPU,
			gpu_percent = 1
	)

Definition at line 58 of file deeplab_segmenter.py.

Member Function Documentation

def segmentation.deeplab_segmenter.DeepLabSegmenter.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 92 of file deeplab_segmenter.py.

def segmentation.deeplab_segmenter.DeepLabSegmenter.run_processed_image	(		self,
			image
	)

Runs inference on a single image.

Args:
image: A PIL.Image object, raw input image.

Returns:
resized_image: RGB image resized from original input image.
seg_map: Segmentation map of `resized_image`.

Definition at line 115 of file deeplab_segmenter.py.

Member Data Documentation

segmentation.deeplab_segmenter.DeepLabSegmenter.graph

Definition at line 66 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.hypes

Definition at line 59 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.input_image_size

Definition at line 85 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.input_tensor

Definition at line 60 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.output_image_uncropped

Definition at line 90 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.output_tensor

Definition at line 61 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.sess

Definition at line 81 of file deeplab_segmenter.py.

segmentation.deeplab_segmenter.DeepLabSegmenter.tools

Definition at line 89 of file deeplab_segmenter.py.

---

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

• deeplab_segmenter.py