欢迎加群:1012878218,一起学习、交流强化学习,里面会有关于深度学习、机器学习、强化学习的各种资料 。
我们对UNet进行了改进,使用ResNet-50的卷积神经网络进行下采样,这样带来的好处有以下几点:1,加深网络层数,提高了网络的分割精度;2,网络中间可以添加更多的跳转连接,这样就能更好的结合图像的背景语义信息,进行多尺度的分割;3,ResNet具有快速收敛,减小模型数据量的优势;4,ResNet使得模型更加的容易训练,既能防止模型退化,又能防止梯度消失,Loss不收敛。
模型的最终结构图如图所示:训练代码prototxt文件在我的资源里面。