apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-31 03:46:40 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

[Benchmark] Update benchmark (#488)

Browse Source 
* add paddle_trt in benchmark

* update benchmark in device

* update benchmark
This commit is contained in:
WJJ1995
2022-11-03 11:23:03 +08:00
committed by GitHub
parent 22d60fdadf
commit 1db85e0ad9
4 changed files with 12 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
benchmark/benchmark_ppcls.py
View File

@@ -22,19 +22,9 @@ import GPUtil
import time
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() == 'true':
return True
elif v.lower() == 'false':
return False
else:
raise argparse.ArgumentTypeError('Boolean value expected.')
def parse_arguments():
import argparse
import ast
parser = argparse.ArgumentParser()
parser.add_argument(
"--model", required=True, help="Path of PaddleClas model.")
@@ -64,12 +54,12 @@ def parse_arguments():
help="inference backend, default, ort, ov, trt, paddle, paddle_trt.")
parser.add_argument(
"--enable_trt_fp16",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable fp16 in trt backend")
parser.add_argument(
"--enable_collect_memory_info",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable collect memory info")
args = parser.parse_args()

16
benchmark/benchmark_ppdet.py
View File

@@ -24,19 +24,9 @@ import GPUtil
import time
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() == 'true':
return True
elif v.lower() == 'false':
return False
else:
raise argparse.ArgumentTypeError('Boolean value expected.')
def parse_arguments():
import argparse
import ast
parser = argparse.ArgumentParser()
parser.add_argument(
"--model", required=True, help="Path of PaddleDetection model.")
@@ -66,12 +56,12 @@ def parse_arguments():
help="inference backend, default, ort, ov, trt, paddle, paddle_trt.")
parser.add_argument(
"--enable_trt_fp16",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable fp16 in trt backend")
parser.add_argument(
"--enable_collect_memory_info",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable collect memory info")
args = parser.parse_args()

16
benchmark/benchmark_ppseg.py
View File

@@ -22,19 +22,9 @@ import GPUtil
import time
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() == 'true':
return True
elif v.lower() == 'false':
return False
else:
raise argparse.ArgumentTypeError('Boolean value expected.')
def parse_arguments():
import argparse
import ast
parser = argparse.ArgumentParser()
parser.add_argument(
"--model", required=True, help="Path of PaddleSeg model.")
@@ -64,12 +54,12 @@ def parse_arguments():
help="inference backend, default, ort, ov, trt, paddle, paddle_trt.")
parser.add_argument(
"--enable_trt_fp16",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable fp16 in trt backend")
parser.add_argument(
"--enable_collect_memory_info",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable collect memory info")
args = parser.parse_args()

16
benchmark/benchmark_yolo.py
View File

@@ -24,19 +24,9 @@ import GPUtil
import time
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() == 'true':
return True
elif v.lower() == 'false':
return False
else:
raise argparse.ArgumentTypeError('Boolean value expected.')
def parse_arguments():
import argparse
import ast
parser = argparse.ArgumentParser()
parser.add_argument(
"--model", required=True, help="Path of Yolo onnx model.")
@@ -66,12 +56,12 @@ def parse_arguments():
help="inference backend, default, ort, ov, trt, paddle, paddle_trt.")
parser.add_argument(
"--enable_trt_fp16",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable fp16 in trt backend")
parser.add_argument(
"--enable_collect_memory_info",
type=str2bool,
type=ast.literal_eval,
default=False,
help="whether enable collect memory info")
args = parser.parse_args()