摘要: import os path = '' path_list=os.listdir(path) for filename in path_list: if os.path.splitext(filename)[1] == '.mhd': XXXXXX 阅读全文
posted @ 2018-07-24 10:43 0点618 阅读(3683) 评论(0) 推荐(0) 编辑
摘要: keras.backend.set_learning_phase(0) # train mode keras.backend.set_learning_phase(1) # predict mode 之所以这么区分,是因为某些层在预测和训练时不同 阅读全文
posted @ 2018-07-24 10:17 0点618 阅读(5511) 评论(1) 推荐(0) 编辑
摘要: 我的环境: win10, 1080ti, cuda=8.0, cudnn=6.0, keras=2.1.2, tensorflow-gpu=1.3 这几天弄这个问题搞得头都大了,重装过显卡驱动,cuda, cudnn各种版本都试了,网上也找不到有效的解决办法。 最后因为在别的win10电脑上程序可以 阅读全文
posted @ 2018-07-24 08:38 0点618 阅读(972) 评论(0) 推荐(0) 编辑