未验证 提交 6ddd8049 编写于 作者: T Tingquan Gao 提交者: GitHub

Add "cpu_num_threads" and "enable_profile" (#494)

上级 c2aed7c0
...@@ -33,6 +33,8 @@ def read_params(): ...@@ -33,6 +33,8 @@ def read_params():
cfg.gpu_mem = 8000 cfg.gpu_mem = 8000
cfg.use_fp16 = False cfg.use_fp16 = False
cfg.use_tensorrt = False cfg.use_tensorrt = False
cfg.cpu_num_threads = 10
cfg.enable_profile = False
# params for preprocess # params for preprocess
cfg.resize_short = 256 cfg.resize_short = 256
......
...@@ -57,19 +57,27 @@ def main(url, image_path, top_k=1): ...@@ -57,19 +57,27 @@ def main(url, image_path, top_k=1):
all_acc = 0.0 all_acc = 0.0
for image_file in image_file_list: for image_file in image_file_list:
file_str = image_file.split('/')[-1]
img = open(image_file, 'rb').read() img = open(image_file, 'rb').read()
if img is None: if img is None:
logger.info("error in loading image:{}".format(image_file)) logger.error("Loading image:{} failed".format(image_file))
continue continue
data = {'images': [cv2_to_base64(img)], 'top_k': top_k} data = {'images': [cv2_to_base64(img)], 'top_k': top_k}
starttime = time.time() starttime = time.time()
try:
r = requests.post(url=url, headers=headers, data=json.dumps(data)) r = requests.post(url=url, headers=headers, data=json.dumps(data))
assert r.status_code == 200, "Request error, status_code: {}".format( r.raise_for_status()
r.status_code) except Exception as e:
logger.error("File:{}, {}".format(file_str, e))
continue
elapse = time.time() - starttime elapse = time.time() - starttime
total_time += elapse total_time += elapse
if r.json()['status'] != '0':
logger.error(
"File:{}, The parameters returned by the server are: {}".
format(file_str, r.json()['msg']))
continue
res = r.json()["results"][0] res = r.json()["results"][0]
classes = res[0] classes = res[0]
scores = res[1] scores = res[1]
...@@ -79,7 +87,6 @@ def main(url, image_path, top_k=1): ...@@ -79,7 +87,6 @@ def main(url, image_path, top_k=1):
scores = map(lambda x: round(x, 5), scores) scores = map(lambda x: round(x, 5), scores)
results = dict(zip(classes, scores)) results = dict(zip(classes, scores))
file_str = image_file.split('/')[-1]
message = "No.{}, File:{}, The top-{} result(s):{}, Time cost:{:.3f}".format( message = "No.{}, File:{}, The top-{} result(s):{}, Time cost:{:.3f}".format(
cnt, file_str, top_k, results, elapse) cnt, file_str, top_k, results, elapse)
logger.info(message) logger.info(message)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册