Spaces:
Running
Running
| 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 | |