对mnist数据集解包出来的图片集,类似这样的
**这是思路的初步代码没写好的,因为不懂怎样将文件夹路径传入类方法中,只能看着网上一些其他方法写写(初学python基础很弱)
**
想问问怎样才能实现上述所说的将文件夹路径传入类方法中并且完成这两个类方法
对mnist数据集解包出来的图片集,类似这样的
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']
如有帮助,请点采纳按钮。