"...composable_kernel_onnx.git" did not exist on "d3051d75175268ee8d6beb64b0177d4c08733291"
Unverified Commit bc462325 authored by MissPenguin's avatar MissPenguin Committed by GitHub
Browse files

Merge pull request #236 from xxxpsyduck/develop

If img is None then .shape method will raise an exception 
parents 1816241d 98c80f16
...@@ -97,11 +97,8 @@ class EvalTestReader(object): ...@@ -97,11 +97,8 @@ class EvalTestReader(object):
if img is None: if img is None:
logger.info("{} does not exist!".format(img_path)) logger.info("{} does not exist!".format(img_path))
continue continue
if len(list(img.shape)) == 2 or img.shape[2] == 1: elif len(list(img.shape)) == 2 or img.shape[2] == 1:
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR) img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
if img is None:
logger.info("load image error:" + img_path)
continue
outs = process_function(img) outs = process_function(img)
outs.append(img_path) outs.append(img_path)
batch_outs.append(outs) batch_outs.append(outs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment