Exemplos de train em Python

Exemplo n.º 1

Arquivo: flowers_train.py Projeto: 1206lyp/models

def main(_):
  dataset = FlowersData(subset=FLAGS.subset)
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.train_dir):
    tf.gfile.DeleteRecursively(FLAGS.train_dir)
  tf.gfile.MakeDirs(FLAGS.train_dir)
  inception_train.train(dataset)

Exemplo n.º 2

def main(_):
    dataset = ImagenetData(subset=FLAGS.subset)
    assert dataset.data_files()
    if tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.DeleteRecursively(FLAGS.train_dir)
    tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_train.train(dataset)

Exemplo n.º 3

def main(unused_argv=None):
  dataset = CamelyonData(subset=FLAGS.subset)
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.train_dir):
    tf.gfile.DeleteRecursively(FLAGS.train_dir)
  tf.gfile.MakeDirs(FLAGS.train_dir)
  inception_train.train(dataset)

Exemplo n.º 4

Arquivo: imagenet_train.py Projeto: mclumd/baxNet

def main(_):
  tf.set_random_seed(1234567)
  dataset = ImagenetData(subset=FLAGS.subset)
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.train_dir):
    tf.gfile.DeleteRecursively(FLAGS.train_dir)
  tf.gfile.MakeDirs(FLAGS.train_dir)
  inception_train.train(dataset)

Exemplo n.º 5

Arquivo: cifar10_train.py Projeto: feifeibear/dist-tensorflow

def main(_):
    dataset = Cifar10Data(subset=FLAGS.subset)
    assert dataset.data_files()
    if tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.DeleteRecursively(FLAGS.train_dir)
    tf.gfile.MakeDirs(FLAGS.train_dir)
    FLAGS.dataset_name = 'cifar10'
    inception_train.train(dataset)

Exemplo n.º 6

def main(_):
    print('===================================')
    print(FLAGS.subset)
    dataset = PlaceData(subset=FLAGS.subset)
    assert dataset.data_files()
    if tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.DeleteRecursively(FLAGS.train_dir)
    tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_train.train(dataset)

Exemplo n.º 7

Arquivo: flowers_train.py Projeto: icodingc/FinalProject

def main(_):
    dataset = FlowersData(subset=FLAGS.subset)
    assert dataset.data_files()
    if tf.gfile.Exists(FLAGS.train_dir):
        print('Dir already exits...')
        sys.exit()
        tf.gfile.DeleteRecursively(FLAGS.train_dir)
    tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_train.train(dataset)

Exemplo n.º 8

Arquivo: flowers_train.py Projeto: jwhpryor/models

def main(_):
  dataset = FlowersData(subset='train')
  validation_dataset = FlowersData(subset='validation')
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.train_dir):
    print('WARNING:  About to delete active train directory...')
    time.sleep(3)
    tf.gfile.DeleteRecursively(FLAGS.train_dir)
  tf.gfile.MakeDirs(FLAGS.train_dir)
  inception_train.train(dataset, validation_dataset)

Exemplo n.º 9

Arquivo: distracted_train.py Projeto: jeremy-poorhouse/models

def main(_):
  dataset = DistractedData(subset=FLAGS.subset, examples=int(FLAGS.examples))
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.train_dir):
    tf.gfile.DeleteRecursively(FLAGS.train_dir)
  tf.gfile.MakeDirs(FLAGS.train_dir)

  if FLAGS.model == 'inception':
  	inception_train.train(dataset)
  elif FLAGS.model == 'vgg':
  	vgg_train.train(dataset)
  else:
  	assert False, "%s model not defined" % (FLAGS.model)

Exemplo n.º 10

def main(_):
    csvFileList, csvEvalFileList = select_csv()
    inception_train.train(csvFileList)

Exemplo n.º 11

Arquivo: cv_train.py Projeto: wooppy-vcs/WooppyProjectC

def main(_):
    dataset = CVData(subset=FLAGS.subset)
    assert dataset.data_files()
    if not tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_train.train(dataset)