18f9aae4939d79bb3698bb94f46d33c08396daf4,berrynet/engine/tflite_engine.py,,main,#,225

Before Change

                     top_k = args.number_top,
                     num_threads = args.number_threads)
        import sys
        sys.exit(0)
    elif args.engine == "detector":
        engine = TFLiteDetectorEngine(
                     model = args.model,
                     labels = args.labels)

After Change

        //         label: car  conf: 0.40003547072410583  (698, 122) (724, 152)
        logger.debug("Inference takes {} s".format(time.time() - t))

    if args.engine == "classifier":
        for r in model_outputs["annotations"]:
            logger.debug("label: {0}  conf: {1}".format(
                r["label"],
                r["confidence"]
            ))
    elif args.engine == "detector":
        for r in model_outputs["annotations"]:
            logger.debug("label: {0}  conf: {1}  ({2}, {3}) ({4}, {5})".format(
                r["label"],
                r["confidence"],
                r["left"],
                r["top"],
                r["right"],
                r["bottom"]
            ))
    else:
        raise Exception("Can not get result "
                        "from illegal engine {}".format(args.engine))


if __name__ == "__main__":
    main()

Instances