Add --verbose option to infer_tool (#6770)

Detailed infer results printing overload output and freeze sea_runtool.
Hiding debug messages by default.
This commit is contained in:
Ilya Sharikov
2021-07-26 09:00:56 +03:00
committed by GitHub
parent 0c10950032
commit 33dfcd62c2

View File

@@ -62,10 +62,14 @@ def cli_parser():
parser.add_argument('-d', dest='device', required=True, help='Target device to infer on')
parser.add_argument('-r', dest='out_path', required=True,
help='Dumps results to the output file')
parser.add_argument('-v', '--verbose', dest='verbose', action='store_true',
help='Increase output verbosity')
args = parser.parse_args()
ir_path = args.ir_path
device = args.device
out_path = args.out_path
if args.verbose:
log.getLogger().setLevel(log.DEBUG)
return ir_path, device, out_path
@@ -74,6 +78,6 @@ if __name__ == "__main__":
results = infer(ir_path=ir_path, device=device)
np.savez(out_path, **results)
log.info("Path for inference results: {}".format(out_path))
log.info("Inference results:")
log.info(results)
log.info("SUCCESS!")
log.debug("Inference results:")
log.debug(results)
log.debug("SUCCESS!")