Обновлено для Swift 3, это удобная функция, которая принимает CGContext и возвращает UIImage. Обратите внимание, что вам больше не нужно освобождать контекст, когда вы закончите с ним в Swift 3.
func imageFromContext(_ context: CGContext) -> UIImage? {
guard let cgImage = context.makeImage() else { return nil }
return UIImage.init(cgImage: cgImage)
}