openni2

.gitignore

@@ -87,3 +87,12 @@ ENV/
 
 # Rope project settings
 .ropeproject
+
+# Data and pretrained model
+/data/
+/cache/
+/src/eval/
+/src/cache/
+
+# VSCode
+.vscode

README.md

@@ -1,6 +1,7 @@
 # DeepPrior++: Improving Fast and Accurate 3D Hand Pose Estimation
 
 Author: Markus Oberweger <oberweger@icg.tugraz.at>
+Modified by Sandro Kalbermatter for evaluation purposes
 
 ## Requirements:
   * OS

src/data/transformations.py

@@ -130,7 +130,7 @@ def rotatePoint3D(p1, center, angle_x, angle_y, angle_z):
     :return: rotated point
     """
     pp = p1.copy()
-    pp -= center
+    pp = pp - center
     R = getRotationMatrix(angle_x, angle_y, angle_z)
     pr = numpy.array([pp[0], pp[1], pp[2], 1])
     ps = numpy.dot(R, pr)

src/test_realtimepipeline.py

@@ -21,7 +21,7 @@
 """
 
 import os
-os.environ["THEANO_FLAGS"] = "device=gpu,floatX=float32"
+os.environ["THEANO_FLAGS"] = "device=cuda,floatX=float32"
 
 import numpy
 from data.dataset import NYUDataset, ICVLDataset
@@ -30,7 +30,7 @@
 from net.scalenet import ScaleNetParams, ScaleNet
 from util.realtimehandposepipeline import RealtimeHandposePipeline
 from data.importers import ICVLImporter, NYUImporter, MSRA15Importer
-from util.cameradevice import CreativeCameraDevice, FileDevice
+from util.cameradevice import *
 
 
 __author__ = "Markus Oberweger <oberweger@icg.tugraz.at>"
@@ -54,7 +54,7 @@
     # testSeqs = [Seq2]
 
     di = NYUImporter('../data/NYU/')
-    Seq2 = di.loadSequence('test_1')
+    Seq2 = di.loadSequence('test')
     testSeqs = [Seq2]
 
     # load trained network
@@ -74,5 +74,5 @@
     dev = FileDevice(filenames, di)
 
     # use depth camera
-    # dev = CreativeCameraDevice(mirror=True)
+    dev = DepthSenseCameraDevice(mirror=True)
     rtp.processVideoThreaded(dev)