# Test code for MM24:DERO
# To conduct the embedding:
python main.py --dataset test_embed --mode test_embed --image_size 512 --image_dir '...' --wmat_dir 'results/WatermarkMatrix/w_256.mat' --batch_size 4
# To conduct the extracting:
python main.py --dataset test_accuracy --mode test_accuracy --image_size 512 --wmat_dir 'results/WatermarkMatrix/w_256.mat' --batch_size 1 --image_dir '...' 