In vgg.py, as the forward, will return 4 outputs.
But in WDNet.py line252-line254, only use top3 outputs to compute the l1 loss
And in paper, To calculate L1 perceptual loss, we use the outputs of the relu2 2 layer of a pre-trained VGG-16 network [25] to represent the learned features of Y^o and Y and then compute their L1 difference.
But in this project, compute the MSELoss
In vgg.py, as the forward, will return 4 outputs.
But in WDNet.py line252-line254, only use top3 outputs to compute the l1 loss
And in paper,
To calculate L1 perceptual loss, we use the outputs of the relu2 2 layer of a pre-trained VGG-16 network [25] to represent the learned features of Y^o and Y and then compute their L1 difference.But in this project, compute the
MSELoss