The source code of implementing deep neural network (DNN) for digital watermarking, please refer to: https://arxiv.org/abs/2003.12428
The source code is based on TensorFlow (v1.13), sonnet (v1.35) as well as another repo: dnn-lib.
Note the dataset are from Kaggle: https://www.kaggle.com/alxmamaev/flowers-recognition
You have to prepare the tf records yourself, since too big to upload here.