This is the implementation of StegaStamp in pytorch. The original StegaStamp is implemented using TensorFlow.
(It has not been maintained for a long time.)