First, you'll want to plug in your iPhone to your computer. In iTunes, unsync your photos and resync them in order to optimize them for the new OS. Seems to work like a charm!
Come across any other bugs, issues, or problems after updating your iPhone or iPod Touch to the new iOS 4? Leave them in the comments, and I'll hunt down some solutions!