剛學pytorch，代碼一運行到
for i, (imgs, targets, _, _) in enumerate(dataloader):
就出現問題。按照其他地方說法，把num_workers改為0也沒用。opt.num_workers為4。
dataloader = DataLoader(dataset,
batch_size=batch_size,
num_workers=opt.num_workers,
shuffle=True, # disable rectangular training if True
pin_memory=True,
collate_fn=dataset.collate_fn)
請問怎么改，謝謝。
Traceback (most recent call last):
File "train.py", line 344, in <module>
results = train(
File "train.py", line 208, in train
for i, (imgs, targets, _, _) in enumerate(dataloader):
File "/data/abc/anaconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 435, in __next__
data = self._next_data()
File "/data/abc/anaconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 475, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/data/abc/anaconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/data/abc/anaconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/abc/bankuai/utils/datasets.py", line 288, in __getitem__
img, ratio, padw, padh = letterbox(img, new_shape=shape, mode='square')
File "/home/abc/bankuai/utils/datasets.py", line 384, in letterbox
img=cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=https://bbs.csdn.net/topics/color) # padded square
cv2.error: OpenCV(4.5.1) /tmp/pip-req-build-ms668fyv/opencv/modules/core/src/copy.cpp:1493: error: (-215:Assertion failed) value[0] == value[1] && value[0] == value[2] && value[0] == value[3] in function 'copyMakeBorder'
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/267936.html
標籤：機器視覺
上一篇：QGraphicsView拖拽圖片時如何不露出白色的底
下一篇：UML設計小白問題