Hugging Face's logo

Spaces:
LuyangZ
/
Outline_Embedding
Runtime error

App Files Community
Outline_Embedding
File size: 5,422 Bytes 
01df1d6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
 
"""Configurations for ResNets

"""

from cortex_DIM.nn_modules.encoder import ResnetEncoder, FoldedResnetEncoder


_resnet19_32x32 = dict(
    Encoder=ResnetEncoder,
    conv_before_args=[(64, 3, 2, 1, True, False, 'ReLU', None)],
    res_args=[
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 2, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 1, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 2, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 1, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1)
    ],
    fc_args=[(1024, True, False, 'ReLU')],
    local_idx=4,
    fc_idx=0
)

_foldresnet19_32x32 = dict(
    Encoder=FoldedResnetEncoder,
    crop_size=8,
    conv_before_args=[(64, 3, 2, 1, True, False, 'ReLU', None)],
    res_args=[
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 2, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 1, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 2, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 1, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1)
    ],
    fc_args=[(1024, True, False, 'ReLU')],
    local_idx=6,
    fc_idx=0
)

_resnet34_32x32 = dict(
    Encoder=ResnetEncoder,
    conv_before_args=[(64, 3, 2, 1, True, False, 'ReLU', None)],
    res_args=[
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         2),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 2, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 1, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         5),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 2, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 1, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         2)
    ],
    fc_args=[(1024, True, False, 'ReLU')],
    local_idx=2,
    fc_idx=0
)

_foldresnet34_32x32 = dict(
    Encoder=FoldedResnetEncoder,
    crop_size=8,
    conv_before_args=[(64, 3, 2, 1, True, False, 'ReLU', None)],
    res_args=[
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(64, 1, 1, 0, True, False, 'ReLU', None),
          (64, 3, 1, 1, True, False, 'ReLU', None),
          (64 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         2),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 2, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(128, 1, 1, 0, True, False, 'ReLU', None),
          (128, 3, 1, 1, True, False, 'ReLU', None),
          (128 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         5),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 2, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         1),
        ([(256, 1, 1, 0, True, False, 'ReLU', None),
          (256, 3, 1, 1, True, False, 'ReLU', None),
          (256 * 4, 1, 1, 0, True, False, 'ReLU', None)],
         2)
    ],
    fc_args=[(1024, True, False, 'ReLU')],
    local_idx=12,
    fc_idx=0
)

configs = dict(
    resnet19_32x32=_resnet19_32x32,
    resnet34_32x32=_resnet34_32x32,
    foldresnet19_32x32=_foldresnet19_32x32,
    foldresnet34_32x32=_foldresnet34_32x32
)