add imread & imwrite to save memory.