日记-D43

分割模型

今天测试了一下BraTS21数据集的效果,训练了100个epoch,写了一个测试脚本,测试结果还可以。有一定的提升。

模型 Dice系数_edema 敏感度_edema 特异性_edema HD95_edema Dice系数_non-enhancing 敏感度_non-enhancing 特异性_non-enhancing HD95_non-enhancing Dice系数_enhancing 敏感度_enhancing 特异性_enhancing HD95_enhancing
nnUNet 0.8568 0.8669 0.9993 18.1122 0.7290 0.7396 0.9998 12.7886 0.8626 0.8622 0.9998 9.7722
M 0.8552 0.8649 0.9993 17.9957 0.7296 0.7401 0.9998 12.8102 0.8613 0.8545 0.9998 9.8840
3 0.8588 0.8678 0.9993 17.2147 0.7357 0.7519 0.9998 12.1545 0.8639 0.8588 0.9998 9.4601

回归模型

因为将表型信息做嵌入后送入Swin Transformer后效果反而会更差,所以现在想的是做一个跳跃链接,将现在的网络增加一个输入,将Swin Transformer输出的信息与表型信息直接相加后,再输入到全连接网络,测试是否效果会更好。代码跑通了。还有待训练。

教师节

今天教师节,我们硕士生给牢蒋她俩送了点礼物,每人合30多。早上来的时候博士师姐也给牢蒋买了不少东西。今早上天气挺凉快的,于是就走过来了。走到半路下雨了,不知道他们军训的会不会休息。