| ====================================================================== | |
| Layer (type:depth-idx) Param # | |
| ====================================================================== | |
| smp_UNet -- | |
| ├─ResNetEncoder: 1-1 -- | |
| │ └─Conv2d: 2-1 3,136 | |
| │ └─BatchNorm2d: 2-2 128 | |
| │ └─ReLU: 2-3 -- | |
| │ └─MaxPool2d: 2-4 -- | |
| │ └─Sequential: 2-5 -- | |
| │ │ └─BasicBlock: 3-1 -- | |
| │ │ │ └─Conv2d: 4-1 36,864 | |
| │ │ │ └─BatchNorm2d: 4-2 128 | |
| │ │ │ └─ReLU: 4-3 -- | |
| │ │ │ └─Conv2d: 4-4 36,864 | |
| │ │ │ └─BatchNorm2d: 4-5 128 | |
| │ │ └─BasicBlock: 3-2 -- | |
| │ │ │ └─Conv2d: 4-6 36,864 | |
| │ │ │ └─BatchNorm2d: 4-7 128 | |
| │ │ │ └─ReLU: 4-8 -- | |
| │ │ │ └─Conv2d: 4-9 36,864 | |
| │ │ │ └─BatchNorm2d: 4-10 128 | |
| │ └─Sequential: 2-6 -- | |
| │ │ └─BasicBlock: 3-3 -- | |
| │ │ │ └─Conv2d: 4-11 73,728 | |
| │ │ │ └─BatchNorm2d: 4-12 256 | |
| │ │ │ └─ReLU: 4-13 -- | |
| │ │ │ └─Conv2d: 4-14 147,456 | |
| │ │ │ └─BatchNorm2d: 4-15 256 | |
| │ │ │ └─Sequential: 4-16 -- | |
| │ │ │ │ └─Conv2d: 5-1 8,192 | |
| │ │ │ │ └─BatchNorm2d: 5-2 256 | |
| │ │ └─BasicBlock: 3-4 -- | |
| │ │ │ └─Conv2d: 4-17 147,456 | |
| │ │ │ └─BatchNorm2d: 4-18 256 | |
| │ │ │ └─ReLU: 4-19 -- | |
| │ │ │ └─Conv2d: 4-20 147,456 | |
| │ │ │ └─BatchNorm2d: 4-21 256 | |
| │ └─Sequential: 2-7 -- | |
| │ │ └─BasicBlock: 3-5 -- | |
| │ │ │ └─Conv2d: 4-22 294,912 | |
| │ │ │ └─BatchNorm2d: 4-23 512 | |
| │ │ │ └─ReLU: 4-24 -- | |
| │ │ │ └─Conv2d: 4-25 589,824 | |
| │ │ │ └─BatchNorm2d: 4-26 512 | |
| │ │ │ └─Sequential: 4-27 -- | |
| │ │ │ │ └─Conv2d: 5-3 32,768 | |
| │ │ │ │ └─BatchNorm2d: 5-4 512 | |
| │ │ └─BasicBlock: 3-6 -- | |
| │ │ │ └─Conv2d: 4-28 589,824 | |
| │ │ │ └─BatchNorm2d: 4-29 512 | |
| │ │ │ └─ReLU: 4-30 -- | |
| │ │ │ └─Conv2d: 4-31 589,824 | |
| │ │ │ └─BatchNorm2d: 4-32 512 | |
| │ └─Sequential: 2-8 -- | |
| │ │ └─BasicBlock: 3-7 -- | |
| │ │ │ └─Conv2d: 4-33 1,179,648 | |
| │ │ │ └─BatchNorm2d: 4-34 1,024 | |
| │ │ │ └─ReLU: 4-35 -- | |
| │ │ │ └─Conv2d: 4-36 2,359,296 | |
| │ │ │ └─BatchNorm2d: 4-37 1,024 | |
| │ │ │ └─Sequential: 4-38 -- | |
| │ │ │ │ └─Conv2d: 5-5 131,072 | |
| │ │ │ │ └─BatchNorm2d: 5-6 1,024 | |
| │ │ └─BasicBlock: 3-8 -- | |
| │ │ │ └─Conv2d: 4-39 2,359,296 | |
| │ │ │ └─BatchNorm2d: 4-40 1,024 | |
| │ │ │ └─ReLU: 4-41 -- | |
| │ │ │ └─Conv2d: 4-42 2,359,296 | |
| │ │ │ └─BatchNorm2d: 4-43 1,024 | |
| ├─UnetDecoder: 1-2 -- | |
| │ └─Identity: 2-9 -- | |
| │ └─ModuleList: 2-10 -- | |
| │ │ └─DecoderBlock: 3-9 -- | |
| │ │ │ └─Conv2dReLU: 4-44 -- | |
| │ │ │ │ └─Conv2d: 5-7 1,769,472 | |
| │ │ │ │ └─BatchNorm2d: 5-8 512 | |
| │ │ │ │ └─ReLU: 5-9 -- | |
| │ │ │ └─Attention: 4-45 -- | |
| │ │ │ │ └─Identity: 5-10 -- | |
| │ │ │ └─Conv2dReLU: 4-46 -- | |
| │ │ │ │ └─Conv2d: 5-11 589,824 | |
| │ │ │ │ └─BatchNorm2d: 5-12 512 | |
| │ │ │ │ └─ReLU: 5-13 -- | |
| │ │ │ └─Attention: 4-47 -- | |
| │ │ │ │ └─Identity: 5-14 -- | |
| │ │ └─DecoderBlock: 3-10 -- | |
| │ │ │ └─Conv2dReLU: 4-48 -- | |
| │ │ │ │ └─Conv2d: 5-15 442,368 | |
| │ │ │ │ └─BatchNorm2d: 5-16 256 | |
| │ │ │ │ └─ReLU: 5-17 -- | |
| │ │ │ └─Attention: 4-49 -- | |
| │ │ │ │ └─Identity: 5-18 -- | |
| │ │ │ └─Conv2dReLU: 4-50 -- | |
| │ │ │ │ └─Conv2d: 5-19 147,456 | |
| │ │ │ │ └─BatchNorm2d: 5-20 256 | |
| │ │ │ │ └─ReLU: 5-21 -- | |
| │ │ │ └─Attention: 4-51 -- | |
| │ │ │ │ └─Identity: 5-22 -- | |
| │ │ └─DecoderBlock: 3-11 -- | |
| │ │ │ └─Conv2dReLU: 4-52 -- | |
| │ │ │ │ └─Conv2d: 5-23 110,592 | |
| │ │ │ │ └─BatchNorm2d: 5-24 128 | |
| │ │ │ │ └─ReLU: 5-25 -- | |
| │ │ │ └─Attention: 4-53 -- | |
| │ │ │ │ └─Identity: 5-26 -- | |
| │ │ │ └─Conv2dReLU: 4-54 -- | |
| │ │ │ │ └─Conv2d: 5-27 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-28 128 | |
| │ │ │ │ └─ReLU: 5-29 -- | |
| │ │ │ └─Attention: 4-55 -- | |
| │ │ │ │ └─Identity: 5-30 -- | |
| │ │ └─DecoderBlock: 3-12 -- | |
| │ │ │ └─Conv2dReLU: 4-56 -- | |
| │ │ │ │ └─Conv2d: 5-31 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-32 64 | |
| │ │ │ │ └─ReLU: 5-33 -- | |
| │ │ │ └─Attention: 4-57 -- | |
| │ │ │ │ └─Identity: 5-34 -- | |
| │ │ │ └─Conv2dReLU: 4-58 -- | |
| │ │ │ │ └─Conv2d: 5-35 9,216 | |
| │ │ │ │ └─BatchNorm2d: 5-36 64 | |
| │ │ │ │ └─ReLU: 5-37 -- | |
| │ │ │ └─Attention: 4-59 -- | |
| │ │ │ │ └─Identity: 5-38 -- | |
| │ │ └─DecoderBlock: 3-13 -- | |
| │ │ │ └─Conv2dReLU: 4-60 -- | |
| │ │ │ │ └─Conv2d: 5-39 4,608 | |
| │ │ │ │ └─BatchNorm2d: 5-40 32 | |
| │ │ │ │ └─ReLU: 5-41 -- | |
| │ │ │ └─Attention: 4-61 -- | |
| │ │ │ │ └─Identity: 5-42 -- | |
| │ │ │ └─Conv2dReLU: 4-62 -- | |
| │ │ │ │ └─Conv2d: 5-43 2,304 | |
| │ │ │ │ └─BatchNorm2d: 5-44 32 | |
| │ │ │ │ └─ReLU: 5-45 -- | |
| │ │ │ └─Attention: 4-63 -- | |
| │ │ │ │ └─Identity: 5-46 -- | |
| ├─SegmentationHead: 1-3 -- | |
| │ └─Conv2d: 2-11 290 | |
| │ └─Identity: 2-12 -- | |
| │ └─Activation: 2-13 -- | |
| │ │ └─Identity: 3-14 -- | |
| ====================================================================== | |
| Total params: 14,322,082 | |
| Trainable params: 14,322,082 | |
| Non-trainable params: 0 | |
| ====================================================================== |