Skip to content

Commit

Permalink
'
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexanderDokuchaev committed Nov 15, 2024
1 parent db4868e commit 63fdf19
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -115,7 +115,7 @@ def prepare_validation(model: YOLO, args: Any) -> Tuple[SegmentationValidator, t


def benchmark_performance(model_path, config) -> float:
command = f"benchmark_app -m {model_path} -d CPU -api async -t 30 -shape '[1,3,{config.imgsz},{config.imgsz}]"
command = f"benchmark_app -m {model_path} -d CPU -api async -t 30 -shape '[1,3,{config.imgsz},{config.imgsz}]'"
cmd_output = subprocess.check_output(command, shell=True) # nosec
match = re.search(r"Throughput\: (.+?) FPS", str(cmd_output))
return float(match.group(1))
Expand Down

0 comments on commit 63fdf19

Please sign in to comment.