As I'm writing a song, I focus on the music and try not to let my mind wander too much, but if I do think of any visual moments or references that I think would connect nicely, I'll make a note of it.