forgot to reset img size to 512x512