Yes, that’s what I’ve been trying to use, but I keep getting the dimensions mixed up.
I’ve found another post on here asking the same question, and his solution was to just setup a 4D tensor and populate it directly:
data = torch.zeros([len(faces), 3, 224, 224])
self.face_map = []
i = 0
for (uuid, hwid, file) in faces:
img = Image.open(file)
in_t = self.img_tf(img)
data[i] = in_t
i += 1
Like so. I imagine the “proper” approach would be to use stack instead?