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

+5 -3
View File
@@ -41,9 +41,11 @@ def construct_dataset(clearml_info_string):
with open(yaml_filenames[0]) as f:
dataset_definition = yaml.safe_load(f)
assert set(dataset_definition.keys()).issuperset({"train", "test", "val", "nc", "names"}), (
"The right keys were not found in the yaml file, make sure it at least has the following keys: ('train', 'test', 'val', 'nc', 'names')"
)
assert set(
dataset_definition.keys()
).issuperset(
{"train", "test", "val", "nc", "names"}
), "The right keys were not found in the yaml file, make sure it at least has the following keys: ('train', 'test', 'val', 'nc', 'names')"
data_dict = {
"train": (