python实现类方法读取文件夹中图片数量及读取图片标签

Question

对mnist数据集解包出来的图片集，类似这样的

**这是思路的初步代码没写好的，因为不懂怎样将文件夹路径传入类方法中，只能看着网上一些其他方法写写（初学python基础很弱）
**

想问问怎样才能实现上述所说的将文件夹路径传入类方法中并且完成这两个类方法

Answer 1

1.代码中os.walk(path)改成，os.walk(self.path)
2.第17行语句不正确，会报没有足够的值去拆包错误。
试试改成如下：

import os

class Mnist:
    def __init__(self,path):
        self.path=path
    def __len__(self):
        count=0
        for root,dirs,files in os.walk(self.path):
            for file in files:
                ext=os.path.splitext(file)[-1].lower()
                if ext=='.jpg':
                    count+=1
        return count
    def __getitem__(self):
        li=[]
        for root,dirs,files in os.walk(self.path):
            for d in dirs:
                for f in os.listdir(os.path.join(root,d)):
                    if f.endswith('.jpg'):
                        li.append(os.path.join(d,f).rsplit('\\')[1])
        return li
p='imgs'
M=Mnist(p)
print(M.__len__())
print(M.__getitem__())

运行结果：

15
['10.jpg', '11.jpg', '12.jpg', '13.jpg', '14.jpg', '15.jpg', '1.jpg', '2.jpg', '3.jpg', '4.jpg', '5.jpg', '6.jpg', '7.jpg', '8.jpg', '9.jpg']

如有帮助，请点采纳按钮。