HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Refactor Python code (#13448)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-06-09 02:32:17 +02:00 • committed by GitHub
parent 6a234f3639
commit 1b26838def
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 81 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

44
ultralytics/utils/benchmarks.py
View file

@ -208,9 +208,10 @@ class RF100Benchmark:
return self.ds_names, self.ds_cfg_list
def fix_yaml(self, path):
@staticmethod
def fix_yaml(path):
"""
Function to fix yaml train and val path.
Function to fix YAML train and val path.
Args:
path (str): YAML file path.
@ -245,32 +246,19 @@ class RF100Benchmark:
entries = line.split(" ")
entries = list(filter(lambda val: val != "", entries))
entries = [e.strip("\n") for e in entries]
start_class = False
for e in entries:
if e == "all":
if "(AP)" not in entries:
if "(AR)" not in entries:
# parse all
eval = {}
eval["class"] = entries[0]
eval["images"] = entries[1]
eval["targets"] = entries[2]
eval["precision"] = entries[3]
eval["recall"] = entries[4]
eval["map50"] = entries[5]
eval["map95"] = entries[6]
eval_lines.append(eval)
if e in class_names:
eval = {}
eval["class"] = entries[0]
eval["images"] = entries[1]
eval["targets"] = entries[2]
eval["precision"] = entries[3]
eval["recall"] = entries[4]
eval["map50"] = entries[5]
eval["map95"] = entries[6]
eval_lines.append(eval)
eval_lines.extend(
{
"class": entries[0],
"images": entries[1],
"targets": entries[2],
"precision": entries[3],
"recall": entries[4],
"map50": entries[5],
"map95": entries[6],
}
for e in entries
if e in class_names or (e == "all" and "(AP)" not in entries and "(AR)" not in entries)
)
map_val = 0.0
if len(eval_lines) > 1:
print("There's more dicts")