[ios] 컬러 인식

21 Dec 2023

이번 블로그에서는 iOS 앱에서 컬러를 인식하는 방법에 대해 알아보겠습니다. 다음 내용으로 구성됩니다:

이미지에서 컬러 추출
카메라로 실시간 컬러 인식
Core Image 사용하기
결론

이미지에서 컬러 추출

iOS 앱에서 이미지에서 컬러를 추출하는 방법은 여러 가지가 있습니다. 이미지를 불러와서 특정 좌표의 픽셀 값을 확인하거나, Core Image를 사용하여 이미지를 필터링하고 추출하는 방법이 있습니다. 예를 들어, 다음은 이미지에서 특정 좌표의 색상을 추출하는 방법입니다.

func getColorAtPoint(point: CGPoint, fromImage image: UIImage) -> UIColor? {
    guard let cgImage = image.cgImage else { return nil }

    let pixelData = CGDataProvider(data: cgImage.dataProvider!.data)
    let providerData = CFDataGetBytePtr(pixelData?.data)

    let bytesPerPixel = 4
    let bytesPerRow = bytesPerPixel * cgImage.width
    let byteIndex = Int(point.y) * bytesPerRow + Int(point.x) * bytesPerPixel

    let red = CGFloat(providerData![byteIndex]) / 255.0
    let green = CGFloat(providerData![byteIndex + 1]) / 255.0
    let blue = CGFloat(providerData![byteIndex + 2]) / 255.0

    return UIColor(red: red, green: green, blue: blue, alpha: 1.0)
}

카메라로 실시간 컬러 인식

iOS 앱에서 카메라로 실시간으로 컬러를 인식하는 기능을 구현하려면 AVCaptureSession과 AVCaptureVideoDataOutput를 사용하여 카메라 입력을 가져와야 합니다. 그리고 각 프레임을 처리하여 적절한 컬러 정보를 추출할 수 있습니다.

다음은 카메라로 실시간으로 컬러를 인식하는 간단한 예제입니다.

// AVCaptureSession 및 AVCaptureVideoDataOutput 설정
let captureSession = AVCaptureSession()
captureSession.sessionPreset = .photo

if let captureDevice = AVCaptureDevice.default(for: .video),
   let input = try? AVCaptureDeviceInput(device: captureDevice) {
    captureSession.addInput(input)

    let videoOutput = AVCaptureVideoDataOutput()
    videoOutput.setSampleBufferDelegate(self, queue: DispatchQueue(label: "videoQueue"))
    captureSession.addOutput(videoOutput)

    let previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
    previewLayer.frame = view.frame
    view.layer.addSublayer(previewLayer)

    captureSession.startRunning()
}

// AVCaptureVideoDataOutputSampleBufferDelegate 구현
extension YourViewController: AVCaptureVideoDataOutputSampleBufferDelegate {
    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        guard let pixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }

        // 픽셀 버퍼를 처리하고 컬러를 추출하는 코드
    }
}

Core Image 사용하기

iOS의 Core Image 프레임워크를 사용하면 이미지에서 쉽게 컬러 및 특징점을 감지할 수 있습니다. Core Image는 컬러를 인식하고 추적하는데 매우 강력한 기능을 제공합니다.

예를 들어, Core Image를 사용하여 이미지에서 동일 컬러 영역을 추출하는 코드는 다음과 유사할 수 있습니다.

func extractColorAreas(from image: UIImage) -> [UIColor] {
    guard let ciImage = CIImage(image: image) else { return [] }

    let options = [CIDetectorAccuracy: CIDetectorAccuracyHigh]
    let colorDetector = CIDetector(ofType: CIDetectorTypeColor, context: nil, options: options)

    guard let features = colorDetector?.features(in: ciImage) as? [CIColor] else { return [] }

    return features.map { UIColor(ciColor: $0) }
}

결론

이렇게 iOS 앱에서 컬러를 인식하는 다양한 방법을 알아보았습니다. 이미지에서 컬러를 추출하거나 카메라로 실시간으로 컬러를 인식하는 것은 사용자 경험을 향상시키고 창의적인 기능을 구현하는데 도움이 될 것입니다. Core Image를 활용하면 보다 정교한 컬러 인식 알고리즘을 적용할 수 있으므로, 프로젝트에 맞게 적합한 방법을 선택하여 구현할 수 있을 것입니다.

자세한 내용은 Apple의 공식 문서를 참고하시기 바랍니다.