Inheritance diagram for detection.yolo_detector.YoloDetector:

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 54 of file yolo_detector.py.

Constructor & Destructor Documentation

◆ init()

def detection.yolo_detector.YoloDetector.__init__	(	self,
		model_dir,
		original_image_size,
		tensor_io,
		runCPU,
		gpu_percent = `1`
	)

Definition at line 57 of file yolo_detector.py.

Member Function Documentation

◆ run_model_on_image()

def detection.yolo_detector.YoloDetector.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 yolo_detector.py.

◆ run_processed_image()

def detection.yolo_detector.YoloDetector.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.
detected_classes: Segmentation map of `resized_image`.

Definition at line 102 of file yolo_detector.py.

Member Data Documentation

◆ graph

detection.yolo_detector.YoloDetector.graph

Definition at line 65 of file yolo_detector.py.

◆ hypes

detection.yolo_detector.YoloDetector.hypes

Definition at line 58 of file yolo_detector.py.

◆ input_image_size

detection.yolo_detector.YoloDetector.input_image_size

Definition at line 85 of file yolo_detector.py.

◆ input_tensor

detection.yolo_detector.YoloDetector.input_tensor

Definition at line 59 of file yolo_detector.py.

◆ output_image_uncropped

detection.yolo_detector.YoloDetector.output_image_uncropped

Definition at line 90 of file yolo_detector.py.

◆ output_tensor

detection.yolo_detector.YoloDetector.output_tensor

Definition at line 60 of file yolo_detector.py.

◆ sess

detection.yolo_detector.YoloDetector.sess

Definition at line 80 of file yolo_detector.py.

◆ tools

detection.yolo_detector.YoloDetector.tools

Definition at line 89 of file yolo_detector.py.

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

yolo_detector.py

Public Member Functions

Public Attributes