目录
此项目相关信息
github链接:https://github.com/xorangecheng/GlobalGuidance-Net
论文链接(2021MIA论文):https://www.sciencedirect.com/science/article/pii/S1361841521000359
github提供的代码质量真的感人,我改了一下午,作者上传这种质量的代码良心不会痛吗???
显而易见的错误
1、train.py中log_path设置
改为:
timestamp = datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
log_path = os.path.join(ckpt_path, exp_name, f"{timestamp}.txt") # 日志路径
2、train.py中数据读取
不要用ImageFolder,用ImageFolder2。原因自己去看dataset.py
3、train.py中def train里有一句梯度异常检测的代码,删掉
with autograd.detect_anomaly():
这个是debug时候用的
文件缺失
1、ResNext预训练模型
自己去下载resnext_101_32x4d.pt文件,放在modeling\backbone\resnext路径下面
下载链接:https://drive.google.com/file/d/1dnH-IHwmu9xFPlyndqI6MfF4LvH6JKNQ/view
2、SynchronizedBatchNorm2d
相关的语句都注释掉吧,作者根本没提供
3、dataset文件夹下缺__init__.py文件
自己建一个空白的
很容易失误的地方
数据文件夹的命名很死板,作者也没有写在readme里。如下要求:图像放在images文件夹里,masks放在seg文件夹里,并且masks的文件名为图像文件名加_mask
标签:-%,github,Lesion,代码,py,train,path From: https://www.cnblogs.com/xjl-ultrasound/p/18595880