HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Update yolov8_sahi.py (#15470)

Browse source 
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-08-14 02:43:51 +08:00 • committed by GitHub
parent 265166ae4d
commit d7d3449ec6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
examples/YOLOv8-SAHI-Inference-Video/yolov8_sahi.py
View file

@ -12,11 +12,15 @@ from ultralytics.utils.files import increment_path
from ultralytics.utils.plotting import Annotator, colors from ultralytics.utils.plotting import Annotator, colors
class SahiInference: class SAHIInference:
"""Runs YOLOv8 and SAHI for object detection on video with options to view, save, and track results."""
def __init__(self): def __init__(self):
"""Initializes the SAHIInference class for performing sliced inference using SAHI with YOLOv8 models."""
self.detection_model = None self.detection_model = None
def load_model(self, weights): def load_model(self, weights):
"""Loads a YOLOv8 model with specified weights for object detection using SAHI."""
yolov8_model_path = f"models/{weights}" yolov8_model_path = f"models/{weights}"
download_yolov8s_model(yolov8_model_path) download_yolov8s_model(yolov8_model_path)
self.detection_model = AutoDetectionModel.from_pretrained( self.detection_model = AutoDetectionModel.from_pretrained(
@ -98,5 +102,5 @@ class SahiInference:
if __name__ == "__main__": if __name__ == "__main__":
inference = SahiInference() inference = SAHIInference()
inference.inference(**vars(inference.parse_opt())) inference.inference(**vars(inference.parse_opt()))