|
|
@ -198,7 +198,7 @@ class RF100Benchmark: |
|
|
|
os.mkdir("ultralytics-benchmarks") |
|
|
|
os.mkdir("ultralytics-benchmarks") |
|
|
|
safe_download("https://github.com/ultralytics/assets/releases/download/v0.0.0/datasets_links.txt") |
|
|
|
safe_download("https://github.com/ultralytics/assets/releases/download/v0.0.0/datasets_links.txt") |
|
|
|
|
|
|
|
|
|
|
|
with open(ds_link_txt, "r") as file: |
|
|
|
with open(ds_link_txt) as file: |
|
|
|
for line in file: |
|
|
|
for line in file: |
|
|
|
try: |
|
|
|
try: |
|
|
|
_, url, workspace, project, version = re.split("/+", line.strip()) |
|
|
|
_, url, workspace, project, version = re.split("/+", line.strip()) |
|
|
@ -222,7 +222,7 @@ class RF100Benchmark: |
|
|
|
Args: |
|
|
|
Args: |
|
|
|
path (str): YAML file path. |
|
|
|
path (str): YAML file path. |
|
|
|
""" |
|
|
|
""" |
|
|
|
with open(path, "r") as file: |
|
|
|
with open(path) as file: |
|
|
|
yaml_data = yaml.safe_load(file) |
|
|
|
yaml_data = yaml.safe_load(file) |
|
|
|
yaml_data["train"] = "train/images" |
|
|
|
yaml_data["train"] = "train/images" |
|
|
|
yaml_data["val"] = "valid/images" |
|
|
|
yaml_data["val"] = "valid/images" |
|
|
@ -242,7 +242,7 @@ class RF100Benchmark: |
|
|
|
skip_symbols = ["🚀", "⚠️", "💡", "❌"] |
|
|
|
skip_symbols = ["🚀", "⚠️", "💡", "❌"] |
|
|
|
with open(yaml_path) as stream: |
|
|
|
with open(yaml_path) as stream: |
|
|
|
class_names = yaml.safe_load(stream)["names"] |
|
|
|
class_names = yaml.safe_load(stream)["names"] |
|
|
|
with open(val_log_file, "r", encoding="utf-8") as f: |
|
|
|
with open(val_log_file, encoding="utf-8") as f: |
|
|
|
lines = f.readlines() |
|
|
|
lines = f.readlines() |
|
|
|
eval_lines = [] |
|
|
|
eval_lines = [] |
|
|
|
for line in lines: |
|
|
|
for line in lines: |
|
|
|