| ====================================================================== | |
| Layer (type:depth-idx) Param # | |
| ====================================================================== | |
| smp_UNetPlusPlus -- | |
| ├─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 | |
| ├─UnetPlusPlusDecoder: 1-2 -- | |
| │ └─Identity: 2-9 -- | |
| │ └─ModuleDict: 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 589,824 | |
| │ │ │ │ └─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 442,368 | |
| │ │ │ │ └─BatchNorm2d: 5-24 256 | |
| │ │ │ │ └─ReLU: 5-25 -- | |
| │ │ │ └─Attention: 4-53 -- | |
| │ │ │ │ └─Identity: 5-26 -- | |
| │ │ │ └─Conv2dReLU: 4-54 -- | |
| │ │ │ │ └─Conv2d: 5-27 147,456 | |
| │ │ │ │ └─BatchNorm2d: 5-28 256 | |
| │ │ │ │ └─ReLU: 5-29 -- | |
| │ │ │ └─Attention: 4-55 -- | |
| │ │ │ │ └─Identity: 5-30 -- | |
| │ │ └─DecoderBlock: 3-12 -- | |
| │ │ │ └─Conv2dReLU: 4-56 -- | |
| │ │ │ │ └─Conv2d: 5-31 184,320 | |
| │ │ │ │ └─BatchNorm2d: 5-32 128 | |
| │ │ │ │ └─ReLU: 5-33 -- | |
| │ │ │ └─Attention: 4-57 -- | |
| │ │ │ │ └─Identity: 5-34 -- | |
| │ │ │ └─Conv2dReLU: 4-58 -- | |
| │ │ │ │ └─Conv2d: 5-35 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-36 128 | |
| │ │ │ │ └─ReLU: 5-37 -- | |
| │ │ │ └─Attention: 4-59 -- | |
| │ │ │ │ └─Identity: 5-38 -- | |
| │ │ └─DecoderBlock: 3-13 -- | |
| │ │ │ └─Conv2dReLU: 4-60 -- | |
| │ │ │ │ └─Conv2d: 5-39 147,456 | |
| │ │ │ │ └─BatchNorm2d: 5-40 128 | |
| │ │ │ │ └─ReLU: 5-41 -- | |
| │ │ │ └─Attention: 4-61 -- | |
| │ │ │ │ └─Identity: 5-42 -- | |
| │ │ │ └─Conv2dReLU: 4-62 -- | |
| │ │ │ │ └─Conv2d: 5-43 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-44 128 | |
| │ │ │ │ └─ReLU: 5-45 -- | |
| │ │ │ └─Attention: 4-63 -- | |
| │ │ │ │ └─Identity: 5-46 -- | |
| │ │ └─DecoderBlock: 3-14 -- | |
| │ │ │ └─Conv2dReLU: 4-64 -- | |
| │ │ │ │ └─Conv2d: 5-47 110,592 | |
| │ │ │ │ └─BatchNorm2d: 5-48 128 | |
| │ │ │ │ └─ReLU: 5-49 -- | |
| │ │ │ └─Attention: 4-65 -- | |
| │ │ │ │ └─Identity: 5-50 -- | |
| │ │ │ └─Conv2dReLU: 4-66 -- | |
| │ │ │ │ └─Conv2d: 5-51 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-52 128 | |
| │ │ │ │ └─ReLU: 5-53 -- | |
| │ │ │ └─Attention: 4-67 -- | |
| │ │ │ │ └─Identity: 5-54 -- | |
| │ │ └─DecoderBlock: 3-15 -- | |
| │ │ │ └─Conv2dReLU: 4-68 -- | |
| │ │ │ │ └─Conv2d: 5-55 92,160 | |
| │ │ │ │ └─BatchNorm2d: 5-56 64 | |
| │ │ │ │ └─ReLU: 5-57 -- | |
| │ │ │ └─Attention: 4-69 -- | |
| │ │ │ │ └─Identity: 5-58 -- | |
| │ │ │ └─Conv2dReLU: 4-70 -- | |
| │ │ │ │ └─Conv2d: 5-59 9,216 | |
| │ │ │ │ └─BatchNorm2d: 5-60 64 | |
| │ │ │ │ └─ReLU: 5-61 -- | |
| │ │ │ └─Attention: 4-71 -- | |
| │ │ │ │ └─Identity: 5-62 -- | |
| │ │ └─DecoderBlock: 3-16 -- | |
| │ │ │ └─Conv2dReLU: 4-72 -- | |
| │ │ │ │ └─Conv2d: 5-63 147,456 | |
| │ │ │ │ └─BatchNorm2d: 5-64 128 | |
| │ │ │ │ └─ReLU: 5-65 -- | |
| │ │ │ └─Attention: 4-73 -- | |
| │ │ │ │ └─Identity: 5-66 -- | |
| │ │ │ └─Conv2dReLU: 4-74 -- | |
| │ │ │ │ └─Conv2d: 5-67 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-68 128 | |
| │ │ │ │ └─ReLU: 5-69 -- | |
| │ │ │ └─Attention: 4-75 -- | |
| │ │ │ │ └─Identity: 5-70 -- | |
| │ │ └─DecoderBlock: 3-17 -- | |
| │ │ │ └─Conv2dReLU: 4-76 -- | |
| │ │ │ │ └─Conv2d: 5-71 110,592 | |
| │ │ │ │ └─BatchNorm2d: 5-72 128 | |
| │ │ │ │ └─ReLU: 5-73 -- | |
| │ │ │ └─Attention: 4-77 -- | |
| │ │ │ │ └─Identity: 5-74 -- | |
| │ │ │ └─Conv2dReLU: 4-78 -- | |
| │ │ │ │ └─Conv2d: 5-75 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-76 128 | |
| │ │ │ │ └─ReLU: 5-77 -- | |
| │ │ │ └─Attention: 4-79 -- | |
| │ │ │ │ └─Identity: 5-78 -- | |
| │ │ └─DecoderBlock: 3-18 -- | |
| │ │ │ └─Conv2dReLU: 4-80 -- | |
| │ │ │ │ └─Conv2d: 5-79 73,728 | |
| │ │ │ │ └─BatchNorm2d: 5-80 128 | |
| │ │ │ │ └─ReLU: 5-81 -- | |
| │ │ │ └─Attention: 4-81 -- | |
| │ │ │ │ └─Identity: 5-82 -- | |
| │ │ │ └─Conv2dReLU: 4-82 -- | |
| │ │ │ │ └─Conv2d: 5-83 36,864 | |
| │ │ │ │ └─BatchNorm2d: 5-84 128 | |
| │ │ │ │ └─ReLU: 5-85 -- | |
| │ │ │ └─Attention: 4-83 -- | |
| │ │ │ │ └─Identity: 5-86 -- | |
| │ │ └─DecoderBlock: 3-19 -- | |
| │ │ │ └─Conv2dReLU: 4-84 -- | |
| │ │ │ │ └─Conv2d: 5-87 4,608 | |
| │ │ │ │ └─BatchNorm2d: 5-88 32 | |
| │ │ │ │ └─ReLU: 5-89 -- | |
| │ │ │ └─Attention: 4-85 -- | |
| │ │ │ │ └─Identity: 5-90 -- | |
| │ │ │ └─Conv2dReLU: 4-86 -- | |
| │ │ │ │ └─Conv2d: 5-91 2,304 | |
| │ │ │ │ └─BatchNorm2d: 5-92 32 | |
| │ │ │ │ └─ReLU: 5-93 -- | |
| │ │ │ └─Attention: 4-87 -- | |
| │ │ │ │ └─Identity: 5-94 -- | |
| ├─SegmentationHead: 1-3 -- | |
| │ └─Conv2d: 2-11 290 | |
| │ └─Identity: 2-12 -- | |
| │ └─Activation: 2-13 -- | |
| │ │ └─Identity: 3-20 -- | |
| ====================================================================== | |
| Total params: 15,964,322 | |
| Trainable params: 15,964,322 | |
| Non-trainable params: 0 | |
| ====================================================================== |