befozg commited on
Commit
804fe21
·
1 Parent(s): 921a2d7

added config

Browse files
Files changed (2) hide show
  1. .gitignore +1 -1
  2. 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