Anchor-x/fed-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

完善_parse_yolo_label_file函数的文档字符串，添加参数和返回值说明

Browse Source
This commit is contained in:
Yunhao Meng
2025-10-02 22:34:29 +08:00
parent b19f11125d
commit c2e538898c
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
utils/fed_util.py
View File

@@ -25,6 +25,10 @@ def _parse_yolo_label_file(label_path: str) -> Set[int]:
Return a set of class_ids found in a YOLO .txt label file. Return a set of class_ids found in a YOLO .txt label file.
Empty file -> empty set. Missing file -> empty set. Empty file -> empty set. Missing file -> empty set.
Robust to blank lines / trailing spaces. Robust to blank lines / trailing spaces.
Args:
label_path: path to the label file
Returns:
set of class IDs (integers) found in the file
""" """
class_ids: Set[int] = set() class_ids: Set[int] = set()
if not os.path.exists(label_path): if not os.path.exists(label_path):