Spaces:
Sleeping
Sleeping
File size: 1,401 Bytes
86694c3 |
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 |
from dataclasses import dataclass
# Model architectures
@dataclass
class C:
filters: int
window_size: tuple
strides: tuple
activation: str = "relu"
"""Conv 1 (2 Layers)"""
c1: C = C(38, (13, 26), (13, 26))
c1_layers: list = [c1]
"""Conv 2 (3 Layers)"""
c2_layers: list = [C(35, (6, 7), (5, 6)), C(87, (6, 9), (5, 8))]
"""Conv 3 (4 Layers)"""
c3_layers: list = [C(32, (4, 5), (3, 4)), C(98, (4, 6), (3, 5)), C(128, (4, 6), (3, 5))]
"""Conv 4 (5 Layers)"""
c4_layers: list = [
C(32, (3, 4), (2, 3)),
C(65, (3, 4), (2, 3)),
C(105, (3, 4), (2, 3)),
C(128, (4, 5), (3, 4)),
]
"""Conv 5 (6 Layers)"""
c5_layers: list = [
C(32, (3, 3), (2, 2)),
C(98, (3, 3), (2, 2)),
C(128, (3, 4), (2, 3)),
C(128, (3, 5), (2, 4)),
C(128, (3, 3), (2, 2)),
]
"""Conv 6 (7 Layers)"""
c6_layers: list = [
C(32, (3, 3), (2, 2)),
C(71, (3, 3), (2, 2)),
C(128, (3, 4), (2, 3)),
C(128, (3, 3), (2, 2)),
C(128, (3, 3), (2, 2)),
C(128, (3, 3), (1, 2)),
]
"""Conv 6XL, 7 Layers"""
c6XL_layers: list = [
C(64, (3, 3), (2, 2)),
C(128, (3, 3), (2, 2)),
C(128, (3, 4), (2, 3)),
C(128, (3, 3), (2, 2)),
C(256, (3, 3), (2, 2)),
C(256, (3, 3), (1, 2)),
]
layers_map: dict = {
"C1": c1_layers,
"C2": c2_layers,
"C3": c3_layers,
"C4": c4_layers,
"C5": c5_layers,
"C6": c6_layers,
"C6XL": c6XL_layers,
}
|