From 1d2aa2bfc338fb7bc342b7ee480c1acf3e293979 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Mon, 23 Sep 2024 15:51:13 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/solutions/action_recognition.py | 6 +++--- ultralytics/utils/__init__.py | 1 - 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/ultralytics/solutions/action_recognition.py b/ultralytics/solutions/action_recognition.py index ad3985070..e29716017 100644 --- a/ultralytics/solutions/action_recognition.py +++ b/ultralytics/solutions/action_recognition.py @@ -1,7 +1,7 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license from collections import defaultdict -from typing import List, Optional, Tuple +from typing import List, Tuple import cv2 import numpy as np @@ -45,7 +45,7 @@ class ActionRecognition: ) self.device = select_device(device) - self.fp16 = torch.cuda.is_available() and torch.cuda.is_bf16_supported() and 'cuda' in self.device + self.fp16 = torch.cuda.is_available() and torch.cuda.is_bf16_supported() and "cuda" in self.device # Check if environment supports imshow self.env_check = check_imshow(warn=True) @@ -58,7 +58,7 @@ class ActionRecognition: self.video_classifier = TorchVisionVideoClassifier(video_classifier_model, device=self.device) else: self.video_classifier = HuggingFaceVideoClassifier( - self.labels, model_name=video_classifier_model, device=self.device, fp16= self.fp16 + self.labels, model_name=video_classifier_model, device=self.device, fp16=self.fp16 ) self.track_history = defaultdict(list) diff --git a/ultralytics/utils/__init__.py b/ultralytics/utils/__init__.py index 0fa920052..b5c7fc531 100644 --- a/ultralytics/utils/__init__.py +++ b/ultralytics/utils/__init__.py @@ -1280,7 +1280,6 @@ def crop_and_pad(frame, box, margin_percent): return cv2.resize(square_crop, (224, 224), interpolation=cv2.INTER_LINEAR) - # Run below code on utils init ------------------------------------------------------------------------------------ # Check first-install steps