fix: reverted incorrect merge

This commit is contained in:
JohnMark Sill 2022-02-18 23:01:53 -06:00 committed by GitHub
parent 0ce6e9a249
commit 6425c5b6d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -623,25 +623,6 @@ def load_labels(path, encoding="utf-8"):
else:
return {index: line.strip() for index, line in enumerate(lines)}
def load_labels(path, encoding="utf-8"):
"""Loads labels from file (with or without index numbers).
Args:
path: path to label file.
encoding: label file encoding.
Returns:
Dictionary mapping indices to labels.
"""
with open(path, "r", encoding=encoding) as f:
lines = f.readlines()
if not lines:
return {}
if lines[0].split(" ", maxsplit=1)[0].isdigit():
pairs = [line.split(" ", maxsplit=1) for line in lines]
return {int(index): label.strip() for index, label in pairs}
else:
return {index: line.strip() for index, line in enumerate(lines)}
class FrameManager(ABC):
@abstractmethod