@thohemp I am using the [test](https://github.com/thohemp/6DRepNet360/blob/master/sixdrepnet360/test.py) script to run on my own video. However, the result is not very good as seen below.  could it be that the model available to download is not trained on CMU dataset?