Minor bug fix to only process color images contained in the dir for color swap experiment