| 25 | |
| 26 | |
| 27 | class RRDB(torch.nn.Module): |
| 28 | |
| 29 | def __init__(self, num_feat, num_grow_ch=32): |
| 30 | super(RRDB, self).__init__() |
| 31 | self.rdb1 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 32 | self.rdb2 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 33 | self.rdb3 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 34 | |
| 35 | def forward(self, x): |
| 36 | out = self.rdb1(x) |
| 37 | out = self.rdb2(out) |
| 38 | out = self.rdb3(out) |
| 39 | return out * 0.2 + x |
| 40 | |
| 41 | |
| 42 | class RRDBNet(torch.nn.Module): |