nnUNetV2修改笔记-D22

训练

跑UNet,有点复杂,不行就用以前用过的数据算了。

昨天把孙梓钊师兄的代码改了下,用之前的jpg和xml,然后把网络最后一层改成单输出,做回归。MSE损失直接爆炸,也没有啥变好的意思。

微信图片_20240716155627.png
感觉有点爆炸。然后下了他们19年的论文,看着好像他们也就这样了
微信图片_20240716155835.png
这看着也是好几万。感觉不行的话还改成分类算了,改成分类20类。我看还有有论文写的直接分三类的。有点搞。