关于训练loss值问题

shartoo / luna16_multi_size_3dcnn

An implement of paper "Multi-level Contextual 3D CNNs for False Positive Reduction in Pulmonary Nodule Detection"

114 stars 44 forks source link

Open Leebanana opened 5 years ago

Leebanana commented 5 years ago

不好意思打扰一下，想问下作者之前说的loss值不收敛的问题是否仍然存在，现在这个项目的效果如何，可以用吗？@shartoo

shartoo commented 5 years ago

抱歉，还不能用，我没有GPU了。。没法训练了。

Leebanana commented 5 years ago

@shartoo 那大概存在什么问题呢，是数据处理还是模型问题呢，可以的话我可以试着改改

shartoo commented 5 years ago

模型结构应该是有问题的，你关注下模型的最后两层以及loss函数的衔接是不是有问题。

Leebanana commented 5 years ago

有个问题想问下，之前提取立方体的格式是xyz，为什么模型结构中卷积核的形式是zyx

shartoo commented 5 years ago

提取成npy之后就成了正常格式，再计算是就是正常格式了。