Random downsample experiment

natanielruiz

2017-09-27 8f2b081586161a55388934456a9b0193d02dd267

 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.