deeptracking

Torch + python implementation of Deep 6-DOF Tracking

Generating data

Generating synthetic data

Run:

python generate_synthetic_data.py config_file.json

dependencies

• cv2
• tqdm
• pyOpenGL
• glfw
• plyfile
• numpngw

configuration

see this example file

Generating real data

Run:

python generate_real_data.py config_file.json

dependencies

• cv2
• tqdm
• pyOpenGL
• glfw
• plyfile
• numpngw

configuration

see this example file

Train

Run:

python train.py config_file.json

dependencies

• Hugh Perkins's pytorch
• scipy, skimage, numpy
• tqdm
• plyfile
• numpngw
• slackclient (could be removed)

configuration

see this example file

Test

Sensor

Will run the tracker with a sensor (kinect 2)

python test_sensor.py config_file.json

Sequence

Will run the tracker on a sequence (folder) and save the error in a file

python test_sequence.py config_file.json

dependencies

• cv2
• Hugh Perkins's pytorch
• pyOpenGL
• glfw
• plyfile
• numpngw
• pyfreenect2 (for Kinect2 sensor)

configuration

see this example file