Пример #1
0
    def main(self, args):

        from aetros.predict import predict
        parser = argparse.ArgumentParser(
            formatter_class=argparse.RawTextHelpFormatter,
            prog=aetros.const.__prog__ + ' predict')
        parser.add_argument(
            'id',
            nargs='?',
            help=
            'the job id, e.g. peter/mnist/5d0f81d3ea73e8b2da3236c93f502339190c7037'
        )
        parser.add_argument(
            '--weights',
            help=
            "Weights path. Per default we try to find it in the ./weights/ folder."
        )
        parser.add_argument('-i',
                            nargs='+',
                            help="Input (path or url). Multiple allowed")

        parsed_args = parser.parse_args(args)

        if not parsed_args.id:
            parser.print_help()
            sys.exit()

        if not parsed_args.i:
            parser.print_help()
            sys.exit()

        os.environ['KERAS_BACKEND'] = 'tensorflow'
        predict(self.logger, parsed_args.id, parsed_args.i,
                parsed_args.weights)
Пример #2
0
    def main(self, args):

        from aetros.predict import predict
        parser = argparse.ArgumentParser(
            formatter_class=argparse.RawTextHelpFormatter,
            prog=aetros.const.__prog__ + ' predict')
        parser.add_argument('job',
                            nargs='?',
                            help='the job id, e.g. GoVDO1Njm')
        parser.add_argument('--insights',
                            action='store_true',
                            help="activates insights")
        parser.add_argument(
            '--weights',
            help=
            "Weights path. Per default we try to find it in the ./weights/ folder."
        )
        parser.add_argument('-i',
                            nargs='+',
                            help="Input (path or url). Multiple allowed")

        parsed_args = parser.parse_args(args)

        if not parsed_args.job:
            parser.print_help()
            sys.exit()

        predict(parsed_args.job, parsed_args.i, parsed_args.insights,
                parsed_args.weights)
Пример #3
0
    def main(self, args):

        from aetros.predict import predict
        parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter, prog=aetros.const.__prog__ + ' predict')
        parser.add_argument('id', nargs='?', help='the job id, e.g. peter/mnist/5d0f81d3ea73e8b2da3236c93f502339190c7037')
        parser.add_argument('--weights', help="Weights path. Per default we try to find it in the ./weights/ folder.")
        parser.add_argument('-i', nargs='+', help="Input (path or url). Multiple allowed")

        parsed_args = parser.parse_args(args)

        if not parsed_args.id:
            parser.print_help()
            sys.exit()

        if not parsed_args.i:
            parser.print_help()
            sys.exit()

        os.environ['KERAS_BACKEND'] = 'tensorflow'
        predict(self.logger, parsed_args.id, parsed_args.i, parsed_args.weights)
Пример #4
0
    def main(self, args):

        from aetros.predict import predict
        parser = argparse.ArgumentParser(
            formatter_class=argparse.RawTextHelpFormatter,
            prog=aetros.const.__prog__ + ' predict')
        parser.add_argument('job',
                            nargs='?',
                            help='the job id, e.g. GoVDO1Njm')
        parser.add_argument(
            '--secure-key',
            help="Secure key. Alternatively use API_KEY environment varibale.")
        parser.add_argument('--insights',
                            action='store_true',
                            help="activates insights")
        parser.add_argument(
            '--weights',
            help=
            "Weights path. Per default we try to find it in the ./weights/ folder."
        )
        parser.add_argument('-i',
                            nargs='+',
                            help="Input (path or url). Multiple allowed")
        parser.add_argument('--tf',
                            action='store_true',
                            help="Uses TensorFlow instead of Theano")

        parsed_args = parser.parse_args(args)

        if not parsed_args.job:
            parser.print_help()
            sys.exit()

        os.environ[
            'KERAS_BACKEND'] = 'tensorflow' if parsed_args.tf else 'theano'
        predict(parsed_args.job,
                parsed_args.i,
                parsed_args.insights,
                parsed_args.weights,
                api_token=parsed_args.secure_key)