class

GS::Vision::Detector

Inherits Reference < Object

YOLO object detector

Constructors

new(config : DetectorConfig = DetectorConfig.new)
Source

Class methods

class_name(class_id : Int32) : String

Get class name

Source

Instance methods

detect(image : Tensor) : Array(Detection)

Detect objects in RGB tensor [H, W, 3]

Source
detect(frame : Video::Frame) : Array(Detection)

Detect objects in video frame

Source
detect_vehicles(image : Tensor) : Array(Detection)

Convenience method: detect only vehicles

Source
finalize
Source
load(model_path : String) : Bool

Load YOLO model from path

Source
loaded?

Check if model is loaded

Source
release

Release model resources

Source