Traceback (most recent call last):
File "train.py", line 542, in
train() # train normally
File "train.py", line 412, in train
results, maps = test.test(cfg,
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/test.py", line 64, in test
for batch_i, (imgs, targets, paths, shapes) in enumerate(tqdm(dataloader, desc=s)):
File "/home/fut/miniconda3/lib/python3.8/site-packages/tqdm/std.py", line 1171, in iter
for obj in iterable:
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 435, in next
data = self._next_data()
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1065, in _next_data
return self._process_data(data)
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1111, in _process_data
data.reraise()
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 428, in reraise
raise self.exc_type(msg)
cv2.error: Caught error in DataLoader worker process 1.
Original Traceback (most recent call last):
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "/home/fut/miniconda3/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 "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/utils/datasets.py", line 426, in getitem
img, ratio, padw, padh = letterbox(img, self.batch_shapes[self.batch[index]], mode='rect')
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/utils/datasets.py", line 643, in letterbox
img = cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color) # add border
cv2.error: OpenCV(4.5.1) /tmp/pip-req-build-ms668fyv/opencv/modules/core/src/copy.cpp:1445: error: (-215:Assertion failed) top >= 0 && bottom >= 0 && left >= 0 && right >= 0 && _src.dims() <= 2 in function 'copyMakeBorder'
Traceback (most recent call last): File "train.py", line 542, in
train() # train normally
File "train.py", line 412, in train
results, maps = test.test(cfg,
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/test.py", line 64, in test
for batch_i, (imgs, targets, paths, shapes) in enumerate(tqdm(dataloader, desc=s)):
File "/home/fut/miniconda3/lib/python3.8/site-packages/tqdm/std.py", line 1171, in iter
for obj in iterable:
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 435, in next
data = self._next_data()
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1065, in _next_data
return self._process_data(data)
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1111, in _process_data
data.reraise()
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/_utils.py", line 428, in reraise
raise self.exc_type(msg)
cv2.error: Caught error in DataLoader worker process 1.
Original Traceback (most recent call last):
File "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 198, in _worker_loop
data = fetcher.fetch(index)
File "/home/fut/miniconda3/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 "/home/fut/miniconda3/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/utils/datasets.py", line 426, in getitem
img, ratio, padw, padh = letterbox(img, self.batch_shapes[self.batch[index]], mode='rect')
File "/home/fut/myproj/ebike_person_prune_yolov3_spp_20210318/utils/datasets.py", line 643, in letterbox
img = cv2.copyMakeBorder(img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=color) # add border
cv2.error: OpenCV(4.5.1) /tmp/pip-req-build-ms668fyv/opencv/modules/core/src/copy.cpp:1445: error: (-215:Assertion failed) top >= 0 && bottom >= 0 && left >= 0 && right >= 0 && _src.dims() <= 2 in function 'copyMakeBorder'