they went in any of their four directions

"they would go straight in any one of the four directions that the creatures looked towards" or "their" refers to the wheels.

whatever direction the head faced, they followed

"they went in whatever direction the winged creatures were looking" or "they followed in whatever direction the front wheel went"