Zhicheng Wu

关于运行代码遇到的两个问题

13

作者您好!您的思路独具特色,效果也非常好,我尝试使用您公开的代码进行实验, 在观察您的issue后,我使用了您分享的阶段2数据集进行阶段2的实验,但是我遇到了如下两个问题: 1.您在dataloader3d文件的getitem函数返回的是 output = {'img': A_img, 'label': A_label}, 但是在模型训练的时候使用的是 A_img=train_A_data['A_img'].cuda() A_label=train_A_data['A_label'].cuda() 导致如下报错 Traceback (most recent call last): File "/home/lab312/PycharmProjects/DAR-UNet-main/stage_2_seg_train.py", line 132, in A_img=train_A_data['A_img'].cuda() KeyError: 'A_img' 请问这里是否应该改为 A_img=train_A_data['img'].cuda() A_label=train_A_data['label'].cuda() 2.我尝试将代码改为...