python代码的一些经典写法

输出到控制台

def _progress(count, block_size, total_size):
  sys.stdout.write('\r>> Downloading %s %.1f%%' %
                   (filename,
                    float(count * block_size) / float(total_size) * 100.0))
  sys.stdout.flush()

文件存在与创建

os.path.exists(dest_directory)
os.makedirs

文件操作 2

os.path.basename()
返回path最后的文件名。若path以/或\结尾,那么就会返回空值。

解压tar包

print('Extracting file from ', filepath)
tarfile.open(filepath, 'r:gz').extractall(dest_directory)

导入读取模型 pb 文件到 Graph

with tf.Graph().as_default() as graph:
  model_path = os.path.join(FLAGS.model_dir, model_info['model_file_name'])
  print('Model path: ', model_path)
  with gfile.FastGFile(model_path, 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    bottleneck_tensor, resized_input_tensor = (tf.import_graph_def(
        graph_def,
        name='',
        return_elements=[
            model_info['bottleneck_tensor_name'],
            model_info['resized_input_tensor_name'],
        ]))

TF的 .format 应用

tf.logging.warning(
    'WARNING: Folder {} has more than {} images. Some images will '
    'never be selected.'.format(dir_name, MAX_NUM_IMAGES_PER_CLASS))

你可能感兴趣的:(python&TF-codes)