Euler angle visualization

[khandriod]

The demo file runs the camera. How to make inference on videos and images?

I want below results that was represented in your paper.

Thanks

[thohemp]

Check here: https://github.com/thohemp/6DRepNet/blob/master/utils.py#L54

It's already implemented in the test.py script. Set --show_viz True. This will save the first image of each batch. Change these lines for your convenience. https://github.com/thohemp/6DRepNet/blob/master/test.py#L155

[khandriod]

Thanks. But what is the full command to make inference on videos and images using per-trained model?

[thohemp]

Check the readme.md

python test.py  --batch_size 64 \
                --dataset ALFW2000 \
                --data_dir datasets/AFLW2000 \
                --filename_list datasets/AFLW2000/files.txt \
                --snapshot output/snapshots/1.pth \
                --show_viz False 

[khandriod]

What is filename_list? and is there way to train the model on custom datasets?