added configs

.gitignore

@@ -160,7 +160,7 @@ cython_debug/
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
 
-config/*
+#config/*
 trainer/__pycache__/
 trainer/__pycache__/*
 __pycache__/*

config/base1024.yaml

@@ -0,0 +1,47 @@
+world_size: 4
+experiment_name: 'FFHQH_SEHARM_1024_FINAL'
+datasets:
+    iharmony: "/home/jovyan/datasets/iHarmony4/"
+    crop_size: 1024
+    # dataset_root: "/home/jovyan/datasets/iHarmony4/"
+    ffhqh: "/home/jovyan/datasets/synthetic_ffhq/"
+    dataset_root: "/home/jovyan/datasets/synthetic_ffhq/"
+    ffhq: "/home/jovyan/datasets/ffhq/images1024x1024/"
+    isTrain: True
+task_for_oneformer: "matting"
+datasets_to_use_test: ["FFHQH"]
+#"IhdDataset", "HCOCO", "HFlickr", "Hday2night", "HAdobe5k", "FFHQH"]
+dataset_to_use: "FFHQH"
+pretrained_model: 'microsoft/swin-base-patch4-window7-224' #"microsoft/swinv2-tiny-patch4-window8-256"  #"nielsr/mask2former-swin-base-youtubevis-2021" #"nvidia/mit-b2"
+batch_size: 4
+num_workers: 4
+log_dir: "log"
+checkpoint_dir: "checkpoints"
+load_pretrained: True
+# checkpoint: "/home/jovyan/efremyan/harmonization/best_50.pth"
+# checkpoint: "/home/jovyan/efremyan/harmonization/checkpoints/01_August_13_38/epoch-13.pth"
+checkpoint: "/home/jovyan/efremyan/harmonization/checkpoints/03_August_23_36/epoch-42.pth"
+distributed_addr: "localhost"
+# distributed_port: "12356"
+# image_size: 512
+lr: 1e-5
+epochs: 300
+disable_validation: False
+disable_mixed_precision: False
+log_image_interval: 10
+log_image_number: 4
+save_model_interval: 1
+lambda_losses: 
+    default: 1.
+    Color: 0.
+    PSNR: 1.
+    LPIPS: 0.5
+    L2: 1.
+    L1: 0.
+    softmax: 1e-4
+model:
+    grid_counts: [10, 5, 1]
+    enc_sizes: [3, 16, 32, 64, 128, 256, 512]
+    init_weights: [0.5, 0.5] # weights of local grided image and global harmonized
+    init_value: 0.8
+    skips: True

config/base512.yaml

@@ -0,0 +1,42 @@
+world_size: 4
+batch_size: 4
+num_workers: 4
+experiment_name: 'test'
+datasets:
+    iharmony: "/home/jovyan/datasets/iHarmony4/"
+    crop_size: 512
+    ffhqh: "/home/jovyan/datasets/synthetic_ffhq/"
+    dataset_root: "/home/jovyan/datasets/synthetic_ffhq/"
+    ffhq: "/home/jovyan/datasets/ffhq/images1024x1024/"
+    isTrain: True
+datasets_to_use_test: ["IhdDataset"]
+dataset_to_use_train: "IhdDataset"
+#["IhdDataset", "HCOCO", "HFlickr", "Hday2night", "HAdobe5k", 
+log_dir: "log"
+checkpoint_dir: "checkpoints"
+load_pretrained: True
+checkpoint: "checkpoints/best_pure.pth"
+distributed_addr: "localhost"
+# distributed_port: "12356"
+# image_size: 512
+lr: 1e-4
+epochs: 300
+disable_validation: False
+disable_mixed_precision: False
+log_image_interval: 10
+log_image_number: 4
+save_model_interval: 1
+lambda_losses: 
+    default: 1.
+    Color: 0.
+    PSNR: 1.
+    LPIPS: 0.5
+    L2: 1.
+    L1: 0.
+    softmax: 1e-4
+model:
+    grid_counts: [10, 5, 1]
+    enc_sizes: [3, 16, 32, 64, 128, 256, 512]
+    init_weights: [0.5, 0.5] # weights of local grided image and global harmonized
+    init_value: 0.8
+    skips: True

config/test.yaml

@@ -0,0 +1,11 @@
+grid_counts: [10, 5, 1]
+enc_sizes: [3, 16, 32, 64, 128, 256, 512]
+init_weights: [0.5, 0.5] # weights of local grided image and global harmonized
+init_value: 0.8
+skips: True
+device: 'cpu'
+checkpoint: 
+  harmonizer: "checkpoints/best_pure.pth"
+  matting: "checkpoints/stylematte.pth"
+
+image_size: 1024