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

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

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

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

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']

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

这行path前面要加.self

for root, dirs, files in os.walk(self.path):

如有帮助，请点击我的回答下方的【采纳该答案】按钮帮忙采纳下，谢谢!

您好，我是有问必答小助手，您的问题已经有小伙伴帮您解答，感谢您对有问必答的支持与关注！
PS：问答VIP年卡 【限时加赠：IT技术图书免费领】，了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632