search

Peachypie98 / RivaGAN

RivaGAN: Robust Invisible Video Watermarking with Attention
12 stars 0 forks source link

Checkpoints #4

Closed Ardor-Wu closed 6 months ago

Ardor-Wu commented 6 months ago

Thank you for sharing the code! Do you have trained checkpoints or is there a way to use their official checkpoint, which ends with .onnx? Thanks!

Peachypie98 commented 6 months ago

At the moment, I have only got a .pt file for pretrained model. (Trained in 32-bits)

Ardor-Wu commented 6 months ago

Could you share that checkpoint? I tried to train with the default setting yet the model does not seem to converge well, with ACC 0.5.

Screenshot 2024-04-17 at 10 13 16 AM
Peachypie98 commented 6 months ago

I have updated the repository, which now includes the pre-trained model weight.