grib0ed0v / face_recognition.pytorch Goto Github PK

change the code to below:

Please which of the LFW images were used ? is it All Images or All Images - alligned with deep funelling?

File "D:/PycharmProjects/work/face_recognition.pytorch-develop/train.py", line 123, in train
for i, data in enumerate(train_loader,0):
File "C:\User\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 435, in next
data = self._next_data()
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 1085, in _next_data
return self._process_data(data)
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data\dataloader.py", line 1111, in _process_data
data.reraise()
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch_utils.py", line 428, in reraise
raise self.exc_type(msg)
AssertionError: Caught AssertionError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\AppData\Local\Programs\Python\Python37\lib\site-packages\torch\utils\data_utils\fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\PycharmProjects\work\face_recognition.pytorch-develop\datasets\casia.py", line 58, in getitem
d_size=(200, 200), normalized=True, show=False)
File "D:\PycharmProjects\work\face_recognition.pytorch-develop\utils\face_align.py", line 28, in align
assert len(landmarks) == 10
AssertionError