Batch testing and hopenet training complete

natanielruiz

2017-09-13 c495a0f6b13b794bab9f6e3423d5038ce645d816

 code/test_on_video.py

@@ -141,7 +141,7 @@
        img_shape = img.size()
        img = img.view(1, img_shape[0], img_shape[1], img_shape[2])
        img = Variable(img).cuda(gpu)
        yaw, pitch, roll = model(img)
        yaw, pitch, roll, angles = model(img)

        yaw_predicted = F.softmax(yaw)
        pitch_predicted = F.softmax(pitch)