added config

base.yaml

@@ -0,0 +1,61 @@
+world_size: 1 
+experiment_name: "test"
+datasets:
+    synthetic_fg: "/home/jovyan/datasets/synthetic_psi/"
+    synthetic_animals: "/home/jovyan/datasets/synthetic_psiny/"
+    bg: "/home/jovyan/datasets/matting/background/testval/"
+    ppm100: "/home/jovyan/kvanchiani/stylegan3/PPM-100/image"
+    aim500: "/home/jovyan/datasets/AIM-500"
+    am2k: 
+        train_original: "/home/jovyan/datasets/matting/am-2k/train/original"
+        train_mask: "/home/jovyan/datasets/matting/am-2k/train/mask"
+        background: "/home/jovyan/datasets/matting/am-2k/background/train"
+        validation_original: "/home/jovyan/datasets/matting/am-2k/validation/original/"
+        validation_mask: "/home/jovyan/datasets/matting/am-2k/validation/mask/"
+        validation_trimap: "/home/jovyan/datasets/matting/am-2k/validation/trimap/"
+    tiktok: "/home/jovyan/datasets/tiktokdataset/dataset"
+    p3m10k: "/home/jovyan/datasets/matting/P3M-10k"
+    p3m10k_test:
+        VAL500P:
+            ROOT_PATH: "P3M-500-P/"
+            ORIGINAL_PATH: "P3M-500-P/blurred_image/"
+            MASK_PATH: "P3M-500-P/mask/"
+            TRIMAP_PATH: "P3M-500-P/trimap/"
+            SAMPLE_NUMBER: 500
+        VAL500NP:
+            ROOT_PATH: "P3M-500-NP/"
+            ORIGINAL_PATH: "P3M-500-NP/original_image/"
+            MASK_PATH: "P3M-500-NP/mask/"
+            TRIMAP_PATH: "P3M-500-NP/trimap/"
+            SAMPLE_NUMBER: 500
+    MAX_SIZE_H: 1600
+    MAX_SIZE_W: 1600
+    image_crop: 800
+    max_image_count: 10000
+dataset_to_use: MixedDataset
+pretrained_model: "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"
+checkpoint: "best-89.pth"
+distributed_addr: "localhost"
+distributed_port: "12357"
+image_size: 800
+lr: 1e-7
+epochs: 200
+disable_validation: False
+warmup_steps: 2
+validate_each_epoch: 5
+max_images_for_validation: 500
+disable_mixed_precision: True
+log_image_interval: 500
+log_image_number: 8
+save_model_interval: 10000
+switch: 3
+lambda_losses: 
+    default: 1.
+    Laplassian: 1.
+    Grad: 3.
+    L1: 1.
+    switch: 1e-6

test.py

@@ -25,8 +25,7 @@ from typing import Type, Any, Callable, Union, List, Optional
 import logging 
 import time
 from omegaconf import OmegaConf
-config = OmegaConf.load(os.path.join(os.path.dirname(
-                os.path.abspath(__file__)), "config/base.yaml"))
+config = OmegaConf.load("base.yaml")
 device = "cuda"
 
 def conv3x3(in_planes, out_planes, stride=1):
@@ -981,8 +980,7 @@ def log(str):
 
 if __name__ == '__main__':
     print('*********************************')  
-    config = OmegaConf.load(os.path.join(os.path.dirname(
-        os.path.abspath(__file__)), "config/base.yaml"))
+    config = OmegaConf.load("base.yaml"))
     config=OmegaConf.merge(config,OmegaConf.from_cli())
     print(config)
     model =  MaskForm()