[swift] Swift에서 카메라로 명함 인식하기

12 Dec 2023

swift

먼저, Vision 프레임워크 및 Core ML 모델을 프로젝트에 추가해야 합니다. 그런 다음 카메라에서 입력된 이미지를 처리하고 명함을 식별하고 정보를 추출하는 코드를 작성해야 합니다.

다음은 Vision 프레임워크를 활용하여 카메라로부터 입력된 이미지를 처리하고 명함을 인식하는 Swift 코드의 간단한 예시입니다.

import AVFoundation
import Vision

class BusinessCardScannerViewController: UIViewController, AVCaptureVideoDataOutputSampleBufferDelegate {
    private var session = AVCaptureSession()

    override func viewDidLoad() {
        super.viewDidLoad()

        // 카메라 캡쳐 세션 설정
        session.sessionPreset = .high

        guard let captureDevice = AVCaptureDevice.default(for: .video) else { return }
        guard let input = try? AVCaptureDeviceInput(device: captureDevice) else { return }
        session.addInput(input)

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

        session.startRunning()
    }

    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }

        let request = VNDetectBusinessCardRequest()
        let handler = VNImageRequestHandler(cvPixelBuffer: pixelBuffer, options: [:])

        do {
            try handler.perform([request])
            guard let observations = request.results as? [VNRecognizedObjectObservation] else { return }
            
            for observation in observations {
                // 여기에서 명함의 정보를 추출하고 처리합니다
                // Core ML 모델을 사용하여 정보를 추출할 수도 있습니다
            }
        } catch {
            print(error)
        }
    }
}

위 코드는 AVCaptureSession을 사용하여 카메라로부터 비디오를 캡처하고, VNDetectBusinessCardRequest를 사용하여 명함을 식별하는 것을 보여줍니다. 명함을 식별한 후에는 observations을 통해 명함의 정보를 추출하고 처리할 수 있습니다.

만약 Core ML 모델을 사용하여 명함의 정보를 추출하려면, Core ML 모델을 프로젝트에 추가하고 이를 활용하여 명함의 정보를 추출하는 코드를 작성해야 합니다.

이 예시는 Vision 프레임워크를 사용한 Swift로 카메라로 명함을 인식하는 간단한 예제이며, 실제 프로젝트에 적용하기 위해서는 더 많은 기능과 에러 핸들링이 필요합니다.