From dc061fe7705380a87e49e1ce219f957e9c376f86 Mon Sep 17 00:00:00 2001 From: UltralyticsAssistant Date: Tue, 30 Jul 2024 19:28:51 +0000 Subject: [PATCH] Auto-format by https://ultralytics.com/actions --- ultralytics/solutions/action_recognition.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ultralytics/solutions/action_recognition.py b/ultralytics/solutions/action_recognition.py index b3e9a6967..110645a41 100644 --- a/ultralytics/solutions/action_recognition.py +++ b/ultralytics/solutions/action_recognition.py @@ -177,7 +177,9 @@ class ActionRecognition: % int(self.num_video_sequence_samples * self.vid_stride * (1 - self.video_cls_overlap_ratio)) == 0 ): - crops_batch = torch.cat(crops_to_infer, dim=0) # crops_batch shape: (batch_size, timestep, height, width, channel) + crops_batch = torch.cat( + crops_to_infer, dim=0 + ) # crops_batch shape: (batch_size, timestep, height, width, channel) output_batch = self.video_classifier(crops_batch) pred_labels, pred_confs = self.postprocess(output_batch)