run.py 文件源码

def main():
    default_weights_path = 'caffe_deep_drive_train_iter_35352.caffemodel'
    parser = argparse.ArgumentParser(description=None)
    parser.add_argument('-v', '--verbose', action='count', dest='verbosity', default=0, help='Set verbosity.')
    parser.add_argument('-w', '--weights', default=default_weights_path, help='Path to caffemodel weights file - default is ' + default_weights_path)
    args = parser.parse_args()

    logging.basicConfig()

    if args.verbosity == 0:
        logger.setLevel(logging.INFO)
    elif args.verbosity >= 1:
        logger.setLevel(logging.DEBUG)

    GTAVRunner._kill_competing_procs()

    enforce_version(GTAV_DIR)

    install_dir = utils.get_config()['install_dir']

    runner = GTAVRunner(install_dir, args.weights)
    runner.popen()
    time.sleep(10)  # Give some time for caffe to create a new log file
    tail_caffe_logs()
    while True:
        if utils.processes_are_running(ALL_PROCESS_NAMES):
            if 'GTAV_DEAD_MANS_SNITCH_URL' in os.environ:  # i.e. https://nosnch.in/a69389848a
                # Send heartbeat (for monitoring long running environments)
                logging.info('Sending heartbeat')
                try:
                    urllib2.urlopen(os.environ['GTAV_DEAD_MANS_SNITCH_URL']).read()
                except Exception as e:
                    logging.error('Error sending heartbeat \n' + traceback.format_exc())
        time.sleep(15 * 60)