Swin-Unet 训练自己数据集踩过的坑

 

py", line 160, in return elem_type({key: default_collate([d[key] for d in batch]) for key in elem}) File "C:\ProgramData\Anaconda3\envs\untitled\lib\site-packages\torch\utils\data\_utils\collate.py", line 141, in default_collate return torch.stack(batch, 0, out=out) RuntimeError: stack expects each tensor to be equal size, but got [224, 224] at entry 0 and [448, 448] at entry 1

以及

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.

 

 标签设置一定要和像素对应否则无法生成正确的数据集格式。

数据集制作如果遇到以下这种这种情况无法识别.npz，自己真是找了好久这个问题，是因为train.txt中没错就是写数据集名称的txt文档中多了一个空格！删除最后一行的空格。