Python tf_allow_growth примеры использования

Пример #1

def init():
    util.lock()#detect previous precess has completed or not
    util.set_img_format()#set image format: channels first or channels last
    util.override_keras_directory_iterator_next()
    util.set_classes_from_train_dir()#data_dir: data/sorted/train/
    util.set_samples_info()

    if util.get_keras_backend_name() != 'theano':
        util.tf_allow_growth()

    if not os.path.exists(config.trained_dir):
        os.mkdir(config.trained_dir)

Пример #2

def init():
    util.lock()
    util.set_img_format()
    util.override_keras_directory_iterator_next()
    util.set_classes_from_train_dir()
    util.set_samples_info()

    if util.get_keras_backend_name() != 'theano':
        util.tf_allow_growth()

    if not os.path.exists(config.trained_dir):
        os.mkdir(config.trained_dir)

Пример #3

Файл: train.py Проект: yashobhan/keras-transfer-learning-for-oxford102

def init():
    util.lock()
    util.set_img_format()
    util.override_keras_directory_iterator_next()
    util.set_classes_from_train_dir()
    util.set_samples_info()

    if util.get_keras_backend_name() != 'theano':
        util.tf_allow_growth()

    if not os.path.exists(config.trained_dir):
        os.mkdir(config.trained_dir)

Пример #4

def init():
    util.lock()
    util.set_img_format()
    # Pay extremely attention to the RGB->BGR, the next method is override.
    util.override_keras_directory_iterator_next()
    util.set_classes_from_train_dir()
    util.set_samples_info()

    if util.get_keras_backend_name() != 'theano':
        util.tf_allow_growth()

    if not os.path.exists(config.trained_dir):
        os.mkdir(config.trained_dir)