未验证 提交 a2bcf2c9 编写于 作者: R Riccardo 提交者: GitHub

Add support for remote validation_outputs_data (#650)

Co-authored-by: Nriccardo <rgall@sertiscorp.com>
上级 ee23a69a
......@@ -353,7 +353,9 @@ def validate(platform, model_file, weight_file, input_file, mace_out_file,
output_names = [name for name in output_node.split(',')]
assert len(input_names) == len(input_shapes)
if not isinstance(validation_outputs_data, list):
if os.path.isfile(validation_outputs_data):
if os.path.isfile(validation_outputs_data) or \
validation_outputs_data.startswith("http://") or \
validation_outputs_data.startswith("https://"):
validation_outputs = [validation_outputs_data]
else:
validation_outputs = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册