WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

4.2. Modeling implicit knowledge question? #263

Open violet2422 opened 2 years ago

violet2422 commented 2 years ago

Hello, I was looking at your interesting work recently. When I saw Section 4.2, I had some questions. How is the 3 modeling methods reflected in the code? image I only see the following code :)

class ImplicitA(nn.Module):
    def __init__(self, channel):
        super(ImplicitA, self).__init__()
        self.channel = channel
        self.implicit = nn.Parameter(torch.zeros(1, channel, 1, 1))
        nn.init.normal_(self.implicit, std=.02)

    def forward(self):
        return self.implicit
WongKinYiu commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/figure/implicit_modeling.png

violet2422 commented 2 years ago

https://github.com/WongKinYiu/yolor/blob/main/figure/implicit_modeling.png