nielsr HF staff commited on
Commit
3c919e7
1 Parent(s): 707cfc6

Upload VitMatteForImageMatting

Browse files
Files changed (2) hide show
  1. config.json +53 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_commit_hash": null,
3
+ "architectures": [
4
+ "VitMatteForImageMatting"
5
+ ],
6
+ "backbone_config": {
7
+ "hidden_size": 384,
8
+ "image_size": 512,
9
+ "model_type": "vitdet",
10
+ "num_attention_heads": 6,
11
+ "num_channels": 4,
12
+ "out_features": [
13
+ "stage12"
14
+ ],
15
+ "out_indices": [
16
+ 12
17
+ ],
18
+ "residual_block_indices": [
19
+ 2,
20
+ 5,
21
+ 8,
22
+ 11
23
+ ],
24
+ "use_relative_position_embeddings": true,
25
+ "window_block_indices": [
26
+ 0,
27
+ 1,
28
+ 3,
29
+ 4,
30
+ 6,
31
+ 7,
32
+ 9,
33
+ 10
34
+ ],
35
+ "window_size": 14
36
+ },
37
+ "convstream_hidden_sizes": [
38
+ 48,
39
+ 96,
40
+ 192
41
+ ],
42
+ "fusion_hidden_sizes": [
43
+ 256,
44
+ 128,
45
+ 64,
46
+ 32
47
+ ],
48
+ "hidden_size": 384,
49
+ "initializer_range": 0.02,
50
+ "model_type": "vitmatte",
51
+ "torch_dtype": "float32",
52
+ "transformers_version": null
53
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ec6aed44bc8d8ab7f4d0ff46da3520a534cf5a97a8262404ff6efa9ae33b1e5
3
+ size 103349013