请问您的训练环境是什么

yuanjiteng / GAN-FM

A method used to fuse infrared and visible images.

6 stars 4 forks source link

请问您的训练环境是什么 #2

Open Thanostitan1022 opened 1 year ago

Thanostitan1022 commented 1 year ago

我现在TensorFlow1.13（gpu）至TensorFlow1.15都尝试过了，运行test_one_image.py （用的是repo里面提供的数据集，没有改动代码）却得不到理想的融合图像，只有黑白噪声图片。当使用tf1=1.13(cpu)版本时可以得到理想的融合结果。请问您的具体训练环境是什么呢？

Thanostitan1022 commented 1 year ago

这是我得到的结果。我的机器环境是RTX3070， Windows10, cuda版本11.7. 使用GPU只能得到下面压缩包里的结果。 results.zip