Python runの例

コード例 #1

ファイル: download_and_convert_data.py プロジェクト: imyourm8/FastMaskRCNN

def main(_):
  if not FLAGS.dataset_name:
    raise ValueError('You must supply the dataset name with --dataset_name')
  if not FLAGS.dataset_dir:
    raise ValueError('You must supply the dataset directory with --dataset_dir')

  elif FLAGS.dataset_name == 'coco':
    download_and_convert_coco.run(FLAGS.dataset_dir, FLAGS.dataset_split_name)
  else:
    raise ValueError(
        'dataset_name [%s] was not recognized.' % FLAGS.dataset_dir)

コード例 #2

def main(_):
    if not FLAGS.dataset_name:
        raise ValueError(
            'You must supply the dataset name with --dataset_name')
    if not FLAGS.dataset_dir:
        raise ValueError(
            'You must supply the dataset directory with --dataset_dir')

    elif FLAGS.dataset_name == 'coco':
        download_and_convert_coco.run(FLAGS.dataset_dir)
    else:
        raise ValueError('dataset_name [%s] was not recognized.' %
                         FLAGS.dataset_dir)

コード例 #3

def main(_):
    if not os.path.isdir('./output/mask_rcnn'):
        os.makedirs('./output/mask_rcnn')
    if not FLAGS.dataset_name:
        raise ValueError(
            'You must supply the dataset name with --dataset_name')
    if not FLAGS.dataset_dir:
        raise ValueError(
            'You must supply the dataset directory with --dataset_dir')

    elif FLAGS.dataset_name == 'coco':
        download_and_convert_coco.run(FLAGS.dataset_dir,
                                      FLAGS.dataset_split_name)
    else:
        raise ValueError('dataset_name [%s] was not recognized.' %
                         FLAGS.dataset_dir)