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

◆ init()

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

◆ run_model_on_image()

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.

◆ run_processed_image()

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

◆ graph

segmentation.deeplab_segmenter.DeepLabSegmenter.graph

Definition at line 66 of file deeplab_segmenter.py.

◆ hypes

segmentation.deeplab_segmenter.DeepLabSegmenter.hypes

Definition at line 59 of file deeplab_segmenter.py.

◆ input_image_size

segmentation.deeplab_segmenter.DeepLabSegmenter.input_image_size

Definition at line 85 of file deeplab_segmenter.py.

◆ input_tensor

segmentation.deeplab_segmenter.DeepLabSegmenter.input_tensor

Definition at line 60 of file deeplab_segmenter.py.

◆ output_image_uncropped

segmentation.deeplab_segmenter.DeepLabSegmenter.output_image_uncropped

Definition at line 90 of file deeplab_segmenter.py.

◆ output_tensor

segmentation.deeplab_segmenter.DeepLabSegmenter.output_tensor

Definition at line 61 of file deeplab_segmenter.py.

◆ sess

segmentation.deeplab_segmenter.DeepLabSegmenter.sess

Definition at line 81 of file deeplab_segmenter.py.

◆ tools

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

Public Member Functions

Public Attributes