Decoder always giving an array of 0

river-huang / ARWGAN

This repository is the official PyTorch implementation of ARWGAN: attention-guided robust image watermarking model based on GAN.

MIT License

28 stars 6 forks source link

Decoder always giving an array of 0 #19

Open SVithurabiman opened 1 month ago

SVithurabiman commented 1 month ago

Thank you for your amazing work. However, when I try to use the pretrained model and the code base , the decoded output always gives an array zeros. I have attached the output image below. Your insights on this would be much helpful