# coding: utf8 # Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserve. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import division from __future__ import print_function from __future__ import unicode_literals import argparse import os def get_arguments(): parser = argparse.ArgumentParser() parser.add_argument( "--use_gpu", action="store_true", help="Use gpu or cpu to test.") parser.add_argument( '--example', type=str, help='RoadLine, HumanSeg or ACE2P') return parser.parse_args() class AttrDict(dict): def __init__(self, *args, **kwargs): super(AttrDict, self).__init__(*args, **kwargs) def __getattr__(self, name): if name in self.__dict__: return self.__dict__[name] elif name in self: return self[name] else: raise AttributeError(name) def __setattr__(self, name, value): if name in self.__dict__: self.__dict__[name] = value else: self[name] = value def merge_cfg_from_args(args, cfg): """Merge config keys, values in args into the global config.""" for k, v in vars(args).items(): d = cfg try: value = eval(v) except: value = v if value is not None: cfg[k] = value