after fg

natanielruiz

2017-10-26 f415df3448622f30c3d1eb680596871672b38dac

 code/hopenet.py

@@ -339,7 +339,7 @@
        preangles = torch.cat([yaw, pitch, roll], 1)
        angles.append(preangles)

        return pre_yaw, pre_pitch, pre_roll, angles, sr_output
        return pre_yaw, pre_pitch, pre_roll, angles, sr_y

class Hopenet_new(nn.Module):
    # This is just Hopenet with 3 output layers for yaw, pitch and roll.