GitHub上找到的AnoGAN代码,运行环境是python2.7,TensorFlow还是CPU低版本的,因为已经装了适配其他版本的cuda所以也没有改用GPU,windows的docker也没有镜像能用,因为没有windows版本的NVIDIA驱动能用
总体配置下来能够使用,需要改动一些小问题
1.batch size 改为1这样得到的图片就是单张图而不是许多图片在一张里面
2.图片设置过大跑不起来,cpu立刻100%
3.utils.py里面的函数改一下才能保存,训练中的图片保存在sample文件夹里,训练一定数量保存一次
def imsave(images, size, path):
image = np.squeeze(merge(images, size))
return imageio.imwrite(path, image)
但我不理解的是我训练几次之后突然保存不了了,希望能得到帮助
AnoGAN基于DCGAN TensorFlow生成图片进行图片异常检测
最新推荐文章于 2024-06-15 11:56:23 发布