Inheritance diagram for yolov8_runner.YoloV8Runner:

Public Member Functions
	__init__ (self, checkpoint, **kwargs)

	run_detector (self, image)

Public Attributes
	model

	kwargs

Detailed Description

YOLO 모델을 동작시키기 위한 클래스
args:
    checkpoint (str): 모델 파라미터 저장 경로, engine 파일(TensorRT 모델) 또는 pt 파일(torch 모델)
    kwargs (str): 모델 동작을 위한 기타 파라미터

Definition at line 3 of file yolov8_runner.py.

Constructor & Destructor Documentation

◆ init()

yolov8_runner.YoloV8Runner.__init__	(		self,
			checkpoint,
		**	kwargs )

Definition at line 10 of file yolov8_runner.py.

    def __init__(self, checkpoint, **kwargs):
        self.model = YOLO(checkpoint)
        self.kwargs = kwargs
 

Member Function Documentation

◆ run_detector()

yolov8_runner.YoloV8Runner.run_detector	(		self,
			image )

객체 검출기를 동작시키는 기능
args:
    image (np.ndarray): RGB? BGR? 이미지

Definition at line 14 of file yolov8_runner.py.

    def run_detector(self, image):
        """
        객체 검출기를 동작시키는 기능
        args:
            image (np.ndarray): RGB? BGR? 이미지
        """
        
        results = self.model.predict(source=image, **self.kwargs)
 
        data = {}
        data['det_bboxes'] = results[0].boxes.data.cpu()[:, :5]
        data['det_labels'] = results[0].boxes.cls.cpu()
        
        return data
 

Member Data Documentation

◆ kwargs

yolov8_runner.YoloV8Runner.kwargs

Definition at line 12 of file yolov8_runner.py.

◆ model

yolov8_runner.YoloV8Runner.model

Definition at line 11 of file yolov8_runner.py.

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

smdetect/runner/yolov8_runner.py

Public Member Functions

Public Attributes