Training AFLW

natanielruiz

2017-09-15 8e5d7dbfe49d194b7d0b616307663e9a88fbcd88

 code/hopenet.py

@@ -117,10 +117,12 @@
        pitch = pitch.view(pitch.size(0), 1)
        roll = roll.view(roll.size(0), 1)
        angles = []
        angles.append(torch.cat([yaw, pitch, roll], 1))
        preangles = torch.cat([yaw, pitch, roll], 1)
        angles.append(preangles)

        # angles predicts the residual
        for idx in xrange(self.iter_ref):
            angles.append(self.fc_finetune(torch.cat((angles[-1], x), 1)))
            angles.append(self.fc_finetune(torch.cat((preangles, x), 1)))

        return pre_yaw, pre_pitch, pre_roll, angles