首页 > 解决方案 > 尝试打印图像计数

问题描述

我是 Python 新手,我正在尝试为一个项目启动 CNN。我安装了 gdrive 并尝试从 gdrive 目录下载图像。之后,我试图计算该目录中的图像。这是我的代码:

import pathlib
dataset_dir = "content/drive/My Drive/Species_Samples"
data_dir = tf.keras.utils.get_file('Species_Samples', origin=dataset_dir, untar=True)
data_dir = pathlib.Path(data_dir)

image_count = len(list(data_dir('*/*.png')))
print(image_count)

但是,我收到以下错误。

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-78-e5d9409807d9> in <module>()
----> 1 image_count = len(list(data_dir('*/*.png')))
      2 print(image_count)

TypeError: 'PosixPath' object is not callable

你能帮忙吗?

建议后,我的代码如下所示:

import pathlib
data_dir = pathlib.Path("content/drive/My Drive/Species_Samples/")
count = len(list(data_dir.rglob("*.png")))
print(count)

标签: conv-neural-networkpathlibgdrive

解决方案


您正在尝试 glob 文件,您需要使用以下 glob 方法之一pathlib

import pathlib

data_dir = pathlib.Path("/path/to/dir/")

count = len(list(data_dir.rglob("*.png")))

在这种情况下.rglob是一个递归 glob。


推荐阅读