Update vision demo visualizer script in light of other recent updates.