Github复现之tiramisu(提拉米苏,这里就是FCDenseNet)
链接:https://github.com/bfortuner/pytorch_tiramisu网络原本是多分类的,而且原始数据输入函数都是为CamVid这样的公开数据集定制的,要用自己的数据还是要花点时间的,我这里是改成了自己的数据输入且为二分类。环境CUDA10.0 CUDNN7.6.0 pytorch1.2.0(环境可以先不改,试试看能不能行,换环境太麻烦),显存12G提示:项目提供的网络有三种复杂度FCDenseNet57、FCDenseNet67、FCDenseNet103,详情见/mode
原创
2020-06-01 13:32:02 ·
2041 阅读 ·
18 评论