Spaces:
Running
Running
File size: 2,139 Bytes
8e0f2c0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
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
|