Model for watermark classification implemented with PyTorch - View it on GitHub