cent664 / SSRIW

Tensorflow implementation of 'Robust Image Watermarking based on Cross-Attention and Invariant Domain Learning'
https://arxiv.org/abs/2310.05395
8 stars 0 forks source link

如何配置环境??? #1

Open shiboliu opened 1 month ago

cent664 commented 1 month ago

I used Kaggle TPUs for most of the training. Their default environment with the installed packages should work.