Hugging Face's logo

Spaces:
dawn17
/
Unet
Sleeping

App Files Community
Unet
File size: 1,690 Bytes 
bcc0f94
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
 
# Input (1, 512, 512)
# Output (64, 512, 512)
decoder_config:
  block5: # (1024, 32, 32)
    in_channels: 1024
    kernel_size: 3
    out_channels: 1024
    padding:
    - 1
    - 1
    stride: 1 # (1024, 32, 32)
  block4: # (1024, 32, 32)
    in_channels: 1024
    kernel_size: 2
    out_channels: 512
    padding:
    - 0
    - 1
    stride: 2 # (512, 64, 64)
  block3: # (512, 64, 64)
    in_channels: 512
    kernel_size: 2
    out_channels: 256
    padding:
    - 0
    - 1
    stride: 2 # (256, 128, 128)
  block2: # (256, 128, 128)
    in_channels: 256
    kernel_size: 2
    out_channels: 128
    padding:
    - 0
    - 1
    stride: 2 # (128, 256, 256)
  block1: # (128, 256, 256)
    in_channels: 128
    kernel_size: 2
    out_channels: 64
    padding:
    - 0
    - 1
    stride: 2 # (64, 512, 512)
encoder_config:
  block1: # (1, 512, 512)
    all_padding: true
    in_channels: 1
    maxpool: true
    n_layers: 2
    out_channels: 64 # (64, 256, 256)
  block2: # (64, 256, 256)
    all_padding: true
    in_channels: 64
    maxpool: true
    n_layers: 2
    out_channels: 128 # (128, 128, 128)
  block3: # (128, 128, 128)
    all_padding: true
    in_channels: 128
    maxpool: true
    n_layers: 2
    out_channels: 256 # (256, 64, 64)
  block4: # (256, 64, 64)
    all_padding: true
    in_channels: 256
    maxpool: true
    n_layers: 2
    out_channels: 512 # (512, 32, 32)
  block5: # (512, 32, 32)
    all_padding: true
    in_channels: 512
    maxpool: false
    n_layers: 2
    out_channels: 512 # (512, 32, 32)
  block6: # (512, 32, 32)
    all_padding: true
    in_channels: 512
    maxpool: false
    n_layers: 2
    out_channels: 1024 # (1024, 32, 32)
nclasses: 2