import glob
from PIL import Image
import numpy as np
import cv2
class getFace:
def __init__(self):
self.mData = []
def from_file(self,root_path):
for img_path in glob.glob(root_path + r'\*'):
img = Image.open(img_path)
img = img.convert('L')
img_data = np.array(img, dtype=float)
img_data = img_data.reshape(-1)
self.mData.append(img_data)
self.mData = np.array(self.mData)
return self.mData
def __getitem__(self, item):
data = self.mData[item]
return data
def __len__(self):
return len(self.mData)

这不是list,不能直接对数组使用append