search

thohemp / 6DRepNet

Official Pytorch implementation of 6DRepNet: 6D Rotation representation for unconstrained head pose estimation.
MIT License
550 stars 72 forks source link

fix the RGB order bug in demo.py #54

Open elliottzheng opened 4 months ago

elliottzheng commented 4 months ago

The previous implementation does not actually convert the image to RGB, as Image.from_array assumes RGB input