運行人臉識別代碼Cosface時出現如下問題，這要怎么解決呀! 剛開始學，求解答。
"D:\Program Files\Anaconda\envs\pytorch\python.exe" F:/Python_code/CosFace/CosFace_pytorch-master/main.py
Namespace(batch_size=512, classifier_type='MCP', cuda=False, database='LFW', epochs=30, is_gray=False, log_interval=100, lr=0.1, momentum=0.9, network='sphere20', no_cuda=False, num_class=10752, root_path='', save_path='checkpoint/', step_size=[16000, 24000], train_list='dataset/cleaned_list.txt', weight_decay=0.0005, workers=4)
DataParallel(
(module): sphere(
(layer1): Sequential(
(0): Conv2d(3, 64, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
(1): PReLU(num_parameters=64)
(2): Block(
(conv1): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=64)
(conv2): Conv2d(64, 64, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=64)
)
)
(layer2): Sequential(
(0): Conv2d(64, 128, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
(1): PReLU(num_parameters=128)
(2): Block(
(conv1): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=128)
(conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=128)
)
(3): Block(
(conv1): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=128)
(conv2): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=128)
)
)
(layer3): Sequential(
(0): Conv2d(128, 256, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
(1): PReLU(num_parameters=256)
(2): Block(
(conv1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=256)
(conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=256)
)
(3): Block(
(conv1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=256)
(conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=256)
)
(4): Block(
(conv1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=256)
(conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=256)
)
(5): Block(
(conv1): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=256)
(conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=256)
)
)
(layer4): Sequential(
(0): Conv2d(256, 512, kernel_size=(3, 3), stride=(2, 2), padding=(1, 1))
(1): PReLU(num_parameters=512)
(2): Block(
(conv1): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu1): PReLU(num_parameters=512)
(conv2): Conv2d(512, 512, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
(prelu2): PReLU(num_parameters=512)
)
)
(fc): Linear(in_features=21504, out_features=512, bias=True)
)
)
length of train Database: 455594
Number of Identities: 10752
2020-08-27 19:27:30 Epoch 1 start training
Cannot load image 0000212\086.jpg
Cannot load image 0115010\004.jpg
Cannot load image 0597223\005.jpg
Cannot load image 0004909\022.jpg
Cannot load image 0842770\101.jpg
Cannot load image 0829576\251.jpg
Cannot load image 0004051\035.jpg
Cannot load image 2315497\008.jpg
Cannot load image 0068338\514.jpg
Traceback (most recent call last):
File "F:/Python_code/CosFace/CosFace_pytorch-master/main.py", line 248, in <module>
main()
File "F:/Python_code/CosFace/CosFace_pytorch-master/main.py", line 169, in main
train(train_loader, model, classifier, criterion, optimizer, epoch)
File "F:/Python_code/CosFace/CosFace_pytorch-master/main.py", line 185, in train
for batch_idx, (data, target) in enumerate(train_loader, 1):
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 363, in __next__
data = self._next_data()
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 989, in _next_data
return self._process_data(data)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\dataloader.py", line 1014, in _process_data
data.reraise()
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\_utils.py", line 395, in reraise
raise self.exc_type(msg)
TypeError: Caught TypeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\worker.py", line 185, in _worker_loop
data = fetcher.fetch(index)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\utils\data\_utils\fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "F:\Python_code\CosFace\CosFace_pytorch-master\dataset.py", line 52, in __getitem__
img = self.transform(img)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torchvision\transforms\transforms.py", line 61, in __call__
img = t(img)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torchvision\transforms\transforms.py", line 551, in forward
return F.hflip(img)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torchvision\transforms\functional.py", line 513, in hflip
return F_pil.hflip(img)
File "D:\Program Files\Anaconda\envs\pytorch\lib\site-packages\torchvision\transforms\functional_pil.py", line 29, in hflip
raise TypeError('img should be PIL Image. Got {}'.format(type(img)))
TypeError: img should be PIL Image. Got <class 'NoneType'>
Process finished with exit code 1
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/10177.html