-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmodel_summary.txt
83 lines (82 loc) · 16 KB
/
model_summary.txt
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
Model: "functional"
┏━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━━━━━━━━━━━━━━┓
┃ Layer (type) ┃ Output Shape ┃ Param # ┃ Connected to ┃
┡━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━━━━━━━━━━━━━━┩
│ input_layer (InputLayer) │ (None, 256, 256, 31) │ 0 │ - │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d (Conv2D) │ (None, 256, 256, 64) │ 17,920 │ input_layer[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_1 (Conv2D) │ (None, 256, 256, 64) │ 36,928 │ conv2d[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ max_pooling2d (MaxPooling2D) │ (None, 128, 128, 64) │ 0 │ conv2d_1[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_2 (Conv2D) │ (None, 128, 128, 128) │ 73,856 │ max_pooling2d[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_3 (Conv2D) │ (None, 128, 128, 128) │ 147,584 │ conv2d_2[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ max_pooling2d_1 │ (None, 64, 64, 128) │ 0 │ conv2d_3[0][0] │
│ (MaxPooling2D) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_4 (Conv2D) │ (None, 64, 64, 256) │ 295,168 │ max_pooling2d_1[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_5 (Conv2D) │ (None, 64, 64, 256) │ 590,080 │ conv2d_4[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ max_pooling2d_2 │ (None, 32, 32, 256) │ 0 │ conv2d_5[0][0] │
│ (MaxPooling2D) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_6 (Conv2D) │ (None, 32, 32, 512) │ 1,180,160 │ max_pooling2d_2[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_7 (Conv2D) │ (None, 32, 32, 512) │ 2,359,808 │ conv2d_6[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ max_pooling2d_3 │ (None, 16, 16, 512) │ 0 │ conv2d_7[0][0] │
│ (MaxPooling2D) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_8 (Conv2D) │ (None, 16, 16, 1024) │ 4,719,616 │ max_pooling2d_3[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_9 (Conv2D) │ (None, 16, 16, 1024) │ 9,438,208 │ conv2d_8[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_transpose │ (None, 32, 32, 512) │ 2,097,664 │ conv2d_9[0][0] │
│ (Conv2DTranspose) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ concatenate (Concatenate) │ (None, 32, 32, 1024) │ 0 │ conv2d_transpose[0][0], │
│ │ │ │ conv2d_7[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_10 (Conv2D) │ (None, 32, 32, 512) │ 4,719,104 │ concatenate[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_11 (Conv2D) │ (None, 32, 32, 512) │ 2,359,808 │ conv2d_10[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_transpose_1 │ (None, 64, 64, 256) │ 524,544 │ conv2d_11[0][0] │
│ (Conv2DTranspose) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ concatenate_1 (Concatenate) │ (None, 64, 64, 512) │ 0 │ conv2d_transpose_1[0][0], │
│ │ │ │ conv2d_5[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_12 (Conv2D) │ (None, 64, 64, 256) │ 1,179,904 │ concatenate_1[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_13 (Conv2D) │ (None, 64, 64, 256) │ 590,080 │ conv2d_12[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_transpose_2 │ (None, 128, 128, 128) │ 131,200 │ conv2d_13[0][0] │
│ (Conv2DTranspose) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ concatenate_2 (Concatenate) │ (None, 128, 128, 256) │ 0 │ conv2d_transpose_2[0][0], │
│ │ │ │ conv2d_3[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_14 (Conv2D) │ (None, 128, 128, 128) │ 295,040 │ concatenate_2[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_15 (Conv2D) │ (None, 128, 128, 128) │ 147,584 │ conv2d_14[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_transpose_3 │ (None, 256, 256, 64) │ 32,832 │ conv2d_15[0][0] │
│ (Conv2DTranspose) │ │ │ │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ concatenate_3 (Concatenate) │ (None, 256, 256, 128) │ 0 │ conv2d_transpose_3[0][0], │
│ │ │ │ conv2d_1[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_16 (Conv2D) │ (None, 256, 256, 64) │ 73,792 │ concatenate_3[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_17 (Conv2D) │ (None, 256, 256, 64) │ 36,928 │ conv2d_16[0][0] │
├───────────────────────────────┼───────────────────────────┼─────────────────┼────────────────────────────┤
│ conv2d_18 (Conv2D) │ (None, 256, 256, 1) │ 65 │ conv2d_17[0][0] │
└───────────────────────────────┴───────────────────────────┴─────────────────┴────────────────────────────┘
Total params: 31,047,873 (118.44 MB)
Trainable params: 31,047,873 (118.44 MB)
Non-trainable params: 0 (0.00 B)