bbox_scores[mask_gt] = pd_scores[ind[0], :, ind[1]][mask_gt] # b, max_num_obj, h*w
bbox_scores[mask_gt] = pd_scores[ind[0], :, ind[1]][mask_gt]# b, max_num_obj, h*w
Traceback (most recent call last):
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\engine\validator.py”, line 181, in call
self.loss += model.loss(batch, preds)[1]
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\nn\tasks.py”, line 258, in loss
return self.criterion(preds, batch)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\utils\loss.py”, line 217, in call
_, target_bboxes, target_scores, fg_mask, _ = self.assigner(
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\nn\modules\module.py”, line 1532, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\nn\modules\module.py”, line 1541, in _call_impl
return forward_call(*args, **kwargs)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\torch\utils_contextlib.py”, line 115, in decorate_context
return func(*args, **kwargs)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\utils\tal.py”, line 72, in forward
mask_pos, align_metric, overlaps = self.get_pos_mask(
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\utils\tal.py”, line 94, in get_pos_mask
align_metric, overlaps = self.get_box_metrics(pd_scores, pd_bboxes, gt_labels, gt_bboxes, mask_in_gts * mask_gt)
File “C:\Users\bx\AppData\Local\anaconda3\envs\mw\lib\site-packages\ultralytics\utils\tal.py”, line 113, in get_box_metrics
bbox_scores[mask_gt] = pd_scores[ind[0], :, ind[1]][mask_gt] # b, max_num_obj, h*w
RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA
to enable device-side assertions.
出现上述问题,是因为txt标签文件中的类别和索引不对应,标签中的类别索引值在数据集的yaml文件中不存在,。
更多推荐
所有评论(0)