search

cszn / DnCNN

Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising (TIP, 2017)
https://cszn.github.io/
1.42k stars 534 forks source link

Version of DnCNN that is compatible with tensorflow 2.0 #83

Open tamireiny opened 3 years ago

tamireiny commented 3 years ago

I have run the code TrainingCodes\dncnn\main_train.py as is, and got the error: AttributeError: module 'tensorflow' has no attribute 'placeholder' I tried to place at main_train.py: import tensorflow.compat.v1 as tf tf.disable_v2_behavior()

Alternatively, I run the tf_upgrade_v2 that suppose to make tensor flow version 1.0, compatible with tensor flow version 2.0, But the output code was exactly the same as the input.

Any suggestions to make the keras training supports tensor flow v2.0?

Thanks