xq141839 / DCSAU-Net

Elsevier-CIBM-2023: A deeper and more compact split-attention U-Net for medical image segmentation
https://www.sciencedirect.com/science/article/pii/S0010482523000914
Apache License 2.0
72 stars 16 forks source link

train #24

Closed FRC1221 closed 10 months ago

FRC1221 commented 10 months ago

data_split_csv.py对images划分,那masks需要划分吗,然后训练报错找不到masks IMG_20240122_155517

xq141839 commented 10 months ago

应该是你的image名字和mask图片名字不一样吧,数据加载的py里默认这两个文件名是一样的,就像我readme里写的数据格式那样,你可以改一下数据加载里的文件或者把mask和image的名字对齐一下。

FRC1221 commented 10 months ago

名称是对齐的的 IMG_20240123_123641 请看看我的划分.py和train.py,填的是否正确 IMG_20240123_123738 IMG_20240123_123814

xq141839 commented 10 months ago

哦哦哦我代码里名字会带文件后缀,你这个应该是jpg和png不一样的关系。可以都转化成png或者数据加载代码里改一下

FRC1221 commented 10 months ago

thank,ok