detection.yolo_detector.YoloDetector Class Reference

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

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

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.

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

detection.yolo_detector.YoloDetector.graph

Definition at line 65 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.hypes

Definition at line 58 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.input_image_size

Definition at line 85 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.input_tensor

Definition at line 59 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.output_image_uncropped

Definition at line 90 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.output_tensor

Definition at line 60 of file yolo_detector.py.

detection.yolo_detector.YoloDetector.sess

Definition at line 80 of file yolo_detector.py.

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