我想通过一下代码去除掉空文件夹,只将含有文件的文件夹加入字典。我在windows上测试是完全没问题的,但是在linux下却无法实现。
def load_label(root_dir):
p = Path(root_dir)
labels = {}
i = 0
for name in os.listdir(p):
# Get rid of empty folders
if os.path.getsize(p / name):
labels[name] = i
i += 1
return labels
后面我发现我所使用的ubuntu系统它的空文件夹依然占有4096Bytes,所以没有办法直接使用。改为下面的代码即可。
if os.path.getsize(p / name)>4096:
[blog.csdn.net/weixin_43944305/article/details/103578222]