当前位置: 首页>>代码示例>>Python>>正文


Python imagenet_data.ImagenetData方法代码示例

本文整理汇总了Python中inception.imagenet_data.ImagenetData方法的典型用法代码示例。如果您正苦于以下问题:Python imagenet_data.ImagenetData方法的具体用法?Python imagenet_data.ImagenetData怎么用?Python imagenet_data.ImagenetData使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在inception.imagenet_data的用法示例。


在下文中一共展示了imagenet_data.ImagenetData方法的9个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
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) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:9,代码来源:imagenet_train.py

示例2: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_args):
  assert FLAGS.job_name in ['ps', 'worker'], 'job_name must be ps or worker'

  # Extract all the hostnames for the ps and worker jobs to construct the
  # cluster spec.
  ps_hosts = FLAGS.ps_hosts.split(',')
  worker_hosts = FLAGS.worker_hosts.split(',')
  tf.logging.info('PS hosts are: %s' % ps_hosts)
  tf.logging.info('Worker hosts are: %s' % worker_hosts)

  cluster_spec = tf.train.ClusterSpec({'ps': ps_hosts,
                                       'worker': worker_hosts})
  server = tf.train.Server(
      {'ps': ps_hosts,
       'worker': worker_hosts},
      job_name=FLAGS.job_name,
      task_index=FLAGS.task_id,
      protocol=FLAGS.protocol)

  if FLAGS.job_name == 'ps':
    # `ps` jobs wait for incoming connections from the workers.
    server.join()
  else:
    # `worker` jobs will actually do the work.
    dataset = ImagenetData(subset=FLAGS.subset)
    assert dataset.data_files()
    # Only the chief checks for or creates train_dir.
    if FLAGS.task_id == 0:
      if not tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_distributed_train.train(server.target, dataset, cluster_spec) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:33,代码来源:imagenet_distributed_train.py

示例3: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_argv=None):
  dataset = ImagenetData(subset=FLAGS.subset)
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.eval_dir):
    tf.gfile.DeleteRecursively(FLAGS.eval_dir)
  tf.gfile.MakeDirs(FLAGS.eval_dir)
  inception_eval.evaluate(dataset) 
开发者ID:ringringyi,项目名称:DOTA_models,代码行数:9,代码来源:imagenet_eval.py

示例4: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_args):
  assert FLAGS.job_name in ['ps', 'worker'], 'job_name must be ps or worker'

  # Extract all the hostnames for the ps and worker jobs to construct the
  # cluster spec.
  ps_hosts = FLAGS.ps_hosts.split(',')
  worker_hosts = FLAGS.worker_hosts.split(',')
  tf.logging.info('PS hosts are: %s' % ps_hosts)
  tf.logging.info('Worker hosts are: %s' % worker_hosts)

  cluster_spec = tf.train.ClusterSpec({'ps': ps_hosts,
                                       'worker': worker_hosts})
  server = tf.train.Server(
      {'ps': ps_hosts,
       'worker': worker_hosts},
      job_name=FLAGS.job_name,
      task_index=FLAGS.task_id)

  if FLAGS.job_name == 'ps':
    # `ps` jobs wait for incoming connections from the workers.
    server.join()
  else:
    # `worker` jobs will actually do the work.
    dataset = ImagenetData(subset=FLAGS.subset)
    assert dataset.data_files()
    # Only the chief checks for or creates train_dir.
    if FLAGS.task_id == 0:
      if not tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_distributed_train.train(server.target, dataset, cluster_spec) 
开发者ID:Cyber-Neuron,项目名称:inception_v3,代码行数:32,代码来源:imagenet_distributed_train.py

示例5: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
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) 
开发者ID:awslabs,项目名称:deeplearning-benchmark,代码行数:9,代码来源:imagenet_train.py

示例6: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_args):
  assert FLAGS.job_name in ['ps', 'worker'], 'job_name must be ps or worker'

  # Extract all the hostnames for the ps and worker jobs to construct the
  # cluster spec.
  ps_hosts = FLAGS.ps_hosts.split(',')
  worker_hosts = FLAGS.worker_hosts.split(',')
  tf.logging.info('PS hosts are: %s' % ps_hosts)
  tf.logging.info('Worker hosts are: %s' % worker_hosts)

  cluster_spec = tf.train.ClusterSpec({'ps': ps_hosts,
                                       'worker': worker_hosts})
  server = tf.train.Server(
      {'ps': ps_hosts,
       'worker': worker_hosts},
      job_name=FLAGS.job_name,
      task_index=FLAGS.task_id)

  if FLAGS.job_name == 'ps':
    # `ps` jobs wait for incoming connections from the workers.
    server.join()
  else:
    # `worker` jobs will actually do the work.
    dataset = ImagenetData(subset=FLAGS.subset)
    #assert dataset.data_files()
    # Only the chief checks for or creates train_dir.
    if FLAGS.task_id == 0:
      if not tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_distributed_train.train(server.target, dataset, cluster_spec) 
开发者ID:awslabs,项目名称:deeplearning-benchmark,代码行数:32,代码来源:imagenet_distributed_train.py

示例7: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
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)
  FLAGS.dataset_name = 'imagenet'
  inception_train.train(dataset) 
开发者ID:wenwei202,项目名称:terngrad,代码行数:10,代码来源:imagenet_train.py

示例8: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_args):
  FLAGS.dataset_name = 'imagenet'

  assert FLAGS.job_name in ['ps', 'worker'], 'job_name must be ps or worker'

  # Extract all the hostnames for the ps and worker jobs to construct the
  # cluster spec.
  ps_hosts = FLAGS.ps_hosts.split(',')
  worker_hosts = FLAGS.worker_hosts.split(',')
  tf.logging.info('PS hosts are: %s' % ps_hosts)
  tf.logging.info('Worker hosts are: %s' % worker_hosts)

  cluster_spec = tf.train.ClusterSpec({'ps': ps_hosts,
                                       'worker': worker_hosts})
  sess_config = tf.ConfigProto()
  sess_config.gpu_options.allow_growth = True

  server = tf.train.Server(
      {'ps': ps_hosts,
       'worker': worker_hosts},
      job_name=FLAGS.job_name,
      task_index=FLAGS.task_id,
      config=sess_config)

  if FLAGS.job_name == 'ps':
    # `ps` jobs wait for incoming connections from the workers.
    server.join()
  else:
    # `worker` jobs will actually do the work.
    dataset = ImagenetData(subset=FLAGS.subset)
    assert dataset.data_files()
    # Only the chief checks for or creates train_dir.
    if FLAGS.task_id == 0:
      if not tf.gfile.Exists(FLAGS.train_dir):
        tf.gfile.MakeDirs(FLAGS.train_dir)
    inception_distributed_train.train(server.target, dataset, cluster_spec) 
开发者ID:wenwei202,项目名称:terngrad,代码行数:38,代码来源:imagenet_distributed_train.py

示例9: main

# 需要导入模块: from inception import imagenet_data [as 别名]
# 或者: from inception.imagenet_data import ImagenetData [as 别名]
def main(unused_argv=None):
  dataset = ImagenetData(subset=FLAGS.subset)
  assert dataset.data_files()
  if tf.gfile.Exists(FLAGS.eval_dir):
    tf.gfile.DeleteRecursively(FLAGS.eval_dir)
  tf.gfile.MakeDirs(FLAGS.eval_dir)
  FLAGS.dataset_name = 'imagenet'
  FLAGS.num_examples = dataset.num_examples_per_epoch()
  inception_eval.evaluate(dataset) 
开发者ID:wenwei202,项目名称:terngrad,代码行数:11,代码来源:imagenet_eval.py


注:本文中的inception.imagenet_data.ImagenetData方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。