Python train 예제들

예제 #1

파일: flowers_train.py 프로젝트: 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)

예제 #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)

예제 #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)

예제 #4

파일: imagenet_train.py 프로젝트: 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)

예제 #5

파일: cifar10_train.py 프로젝트: 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)

예제 #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)

예제 #7

파일: flowers_train.py 프로젝트: 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)

예제 #8

파일: flowers_train.py 프로젝트: 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)

예제 #9

파일: distracted_train.py 프로젝트: 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)

예제 #10

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

예제 #11

파일: cv_train.py 프로젝트: 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)