[swift] Swift에서 카메라로 면허증 인식하기

12 Dec 2023

swift

이번에는 Swift를 사용하여 iOS 앱에서 카메라로 면허증을 스캔하고 인식하는 방법에 대해 알아보겠습니다. 면허증을 인식하기 위해서는 iOS에서 제공하는 Vision 프레임워크를 사용할 수 있습니다. Vision 프레임워크는 이미지 및 비디오 처리를 통해 기계 학습 및 컴퓨터 비전 알고리즘을 구현할 수 있도록 지원합니다.

Vision 프레임워크를 이용한 면허증 인식

Vision 프레임워크를 사용하여 면허증을 인식하는 과정은 다음과 같습니다.

카메라에서 비디오 스트림을 캡처합니다.
각 프레임을 Vision 프레임워크를 사용하여 분석합니다.
면허증에서 특정 정보를 추출합니다.

아래는 간단한 예제 코드를 통해 Vision 프레임워크를 사용하여 카메라로 면허증을 스캔하고 인식하는 방법을 보여줍니다.

import UIKit
import AVFoundation
import Vision

class LicenseScanViewController: UIViewController, AVCaptureVideoDataOutputSampleBufferDelegate {

    private let captureSession = AVCaptureSession()
    private let videoDataOutput = AVCaptureVideoDataOutput()
    private var licenseNumber: String?

    override func viewDidLoad() {
        super.viewDidLoad()

        // AVCaptureSession 설정
        captureSession.sessionPreset = .high
        guard let captureDevice = AVCaptureDevice.default(for: .video),
              let input = try? AVCaptureDeviceInput(device: captureDevice)
        else { return }

        if captureSession.canAddInput(input) && captureSession.canAddOutput(videoDataOutput) {
            captureSession.addInput(input)
            captureSession.addOutput(videoDataOutput)
        }

        videoDataOutput.videoSettings = [kCVPixelBufferPixelFormatTypeKey as String: kCVPixelFormatType_32BGRA]
        videoDataOutput.setSampleBufferDelegate(self, queue: DispatchQueue(label: "videoQueue"))

        captureSession.startRunning()

        // Live Preview 설정
        let previewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
        view.layer.insertSublayer(previewLayer, at: 0)
        previewLayer.frame = view.frame
    }

    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        guard let pixelBuffer: CVPixelBuffer = CMSampleBufferGetImageBuffer(sampleBuffer) else { return }
        let request = VNRecognizeTextRequest { request, error in
            guard let observations = request.results as? [VNRecognizedTextObservation] else { return }
            let recognizedText = observations.compactMap { observation in
                observation.topCandidates(1).first?.string
            }.joined(separator: "\n")
            self.licenseNumber = recognizedText
            print(recognizedText)
        }
        request.recognitionLevel = .accurate
        let imageRequestHandler = VNImageRequestHandler(cvPixelBuffer: pixelBuffer, options: [:])
        do {
            try imageRequestHandler.perform([request])
        } catch {
            print(error)
        }
    }
}

위의 코드에서 LicenseScanViewController는 카메라를 사용하여 면허증을 스캔하고 인식하는 뷰 컨트롤러를 나타냅니다. AVCaptureSession을 통해 비디오 캡처 및 라이브 프리뷰를 설정하고, AVCaptureVideoDataOutputSampleBufferDelegate 프로토콜을 준수하여 비디오 프레임을 처리합니다. VNRecognizeTextRequest를 사용하여 텍스트를 인식하고 추출한 결과를 recognizedText 변수에 저장합니다.

참고 자료

위의 예제 코드를 참고하여 Swift를 사용하여 iOS 앱에서 카메라로 면허증을 스캔하고 인식하는 기능을 구현해 보세요!