最后一层是softmax. 这个loss有一个优化，训练的时候，不需要执行softmax操作，直接把logit和目标图片...
最后一层是softmax. 这个loss有一个优化，训练的时候，不需要执行softmax操作，直接把logit和目标图片...
我运行test_predict文件,使用的是你的unet.hdf5文件和测试文件imgs_test.npy,但预测出来的结果可视化后是几乎全白的。 你可以试试我改过之后的代码,应该能解决你的问题:https://github.com/wuyang0329/unetldhl103 commented on Mar 19, 2019 ldhl103...
这个项目使用主流的深度学习框架 Pytorch + UNet来实现,项目的特点是支持训练、分割算法特别轻量化、能够一键执行训练+预测,能够适应分割结构复杂的医学图像。项目提供完整的代码,包括训练 + 预测代码、一键执行脚本、训练好的分割模型权重 (当然也支持自己训练)、项目三方依赖库 (requirements.txt)、训练标注图片、待...
12.25日增加，当训练时的batchsize为1时，模型抖动会很厉害，导致最后的训练结果有很大的运气成分。