Spaces:
Running
Running
added config
Browse files- .gitignore +1 -1
- config/demo.yaml +12 -0
.gitignore
CHANGED
@@ -175,7 +175,7 @@ flagged/
|
|
175 |
.DS_store
|
176 |
|
177 |
|
178 |
-
config/*
|
179 |
trainer/__pycache__/
|
180 |
trainer/__pycache__/*
|
181 |
__pycache__/*
|
|
|
175 |
.DS_store
|
176 |
|
177 |
|
178 |
+
# config/*
|
179 |
trainer/__pycache__/
|
180 |
trainer/__pycache__/*
|
181 |
__pycache__/*
|
config/demo.yaml
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
grid_counts: [10, 5, 1]
|
2 |
+
enc_sizes: [3, 16, 32, 64, 128, 256, 512]
|
3 |
+
init_weights: [0.5, 0.5] # weights of local grided image and global harmonized
|
4 |
+
init_value: 0.8
|
5 |
+
skips: True
|
6 |
+
device: 'cpu'
|
7 |
+
checkpoint:
|
8 |
+
matting_onnx: "checkpoints/stylematte_720.onnx"
|
9 |
+
harmonizer: "checkpoints/best_pure.pth"
|
10 |
+
matting: "checkpoints/stylematte.pth"
|
11 |
+
onnx: False
|
12 |
+
image_size: 1024
|