(self, x)
| 86 | return shallow_feat1, mid_feat1, deep_feat1, shallow_feat2, mid_feat2, deep_feat2 |
| 87 | |
| 88 | def check_image_size(self, x): |
| 89 | _, _, h, w = x.size() |
| 90 | pad_size = 16 |
| 91 | mod_pad_h = (pad_size - h % pad_size) % pad_size |
| 92 | mod_pad_w = (pad_size - w % pad_size) % pad_size |
| 93 | x = F.pad(x, (0, mod_pad_w, 0, mod_pad_h), 'reflect') |
| 94 | return x |
| 95 | |
| 96 | def forward(self, inp_img): |
| 97 |