Fix TFLite Segment inference (#13488)

* Fix TFLite Segment inference

* Auto-format by https://ultralytics.com

---------

Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Mohammed YasinandUltralyticsAssistant authored and GitHub committed 2025-01-12 19:05:20 +01:00
1 parent 86fd1ab270
commit 6420a1db87
5 files changed
+17 -12

No files matched your search

+3 -3
View File
@@ -355,9 +355,9 @@ class LoadImages:
self._new_video(videos[0]) # new video
else:
self.cap = None
assert self.nf > 0, (
f"No images or videos found in {p}. Supported formats are:\nimages: {IMG_FORMATS}\nvideos: {VID_FORMATS}"
)
assert (
self.nf > 0
), f"No images or videos found in {p}. Supported formats are:\nimages: {IMG_FORMATS}\nvideos: {VID_FORMATS}"
def __iter__(self):
"""Initializes iterator by resetting count and returns the iterator object itself."""