Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Conversation

@wr0124
Copy link
Collaborator

@wr0124 wr0124 commented Oct 7, 2025

This PR introduces a new configuration option, --alg_palette_metric_mask, that allows PSNR and SSIM to be computed only within dilated masked regions.
When enabled, metric evaluation focuses exclusively on the inpainted zones — the parts of the image actually generated by the model.

The command line:

python3 -W ignore::UserWarning -W ignore::FutureWarning train.py \
        --dataroot     path/to/vid  \
        --checkpoints_dir   path/to/ckpt   \
        --name  ddpm_debug  \
        --gpu_ids 1  \
        --data_relative_paths   \
        --model_type palette \
        --data_dataset_mode  self_supervised_vid_mask_online  \
        --train_batch_size 1  \
        --test_batch_size  2 \
        --train_iter_size 1  \
        --data_num_threads  1  \
        --train_G_ema \
        --train_optim adamw \
        --G_netG unet_vid   \
        --data_online_creation_rand_mask_A \
        --dataaug_no_rotate \
        --G_diff_n_timestep_train  20  \
        --G_diff_n_timestep_test   10  \
        --output_print_freq 1   \
        --output_display_freq 1 \
        --data_temporal_number_frames  3  \
        --data_temporal_frame_step   1  \
        --data_crop_size 128 \
        --data_load_size 128  \
        --train_compute_metrics_test   \
        --train_metrics_every 1  \
        --train_metrics_list PSNR LPIPS SSIM \
        --with_amp \
        --with_tf32 \
        --data_online_creation_crop_size_A 300 \
        --data_online_creation_crop_size_B  300  \
        --train_save_epoch_freq 1000  \
        --alg_palette_sampling_method ddpm \
        --alg_palette_ddim_eta 0.5 \
        --alg_palette_ddim_num_steps 40 \
        --alg_palette_sampling_method_test  ddpm \
        --alg_palette_sampling_steps_test 10  \
        --train_G_lr 0.00002   \
        --alg_palette_metric_mask  \

@beniz beniz merged commit 05ac64d into jolibrain:master Oct 14, 2025
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants