[swift] Swift에서 카메라로 여권 인식하기

12 Dec 2023

swift

iOS 어플리케이션을 개발하다보면 종종 여권을 스캔하여 사용해야 하는 경우가 있습니다. 스위프트를 사용하여 카메라로 여권을 식별하고 정보를 추출하는 방법에 대해 알아보겠습니다.

AVFoundation 프레임워크를 사용하여 카메라 액세스

우선, AVFoundation 프레임워크를 사용하여 카메라에 액세스해야 합니다. AVCaptureSession, AVCaptureDevice, AVCaptureDeviceInput, AVCaptureVideoPreviewLayer 등의 클래스를 사용하여 카메라 화면을 표시하고 여권을 촬영할 수 있습니다.

import AVFoundation
import UIKit

class PassportScannerViewController: UIViewController, AVCaptureVideoDataOutputSampleBufferDelegate {
    var captureSession: AVCaptureSession!
    var stillImageOutput: AVCapturePhotoOutput!
    var videoPreviewLayer: AVCaptureVideoPreviewLayer!

    override func viewDidLoad() {
        super.viewDidLoad()
        
        captureSession = AVCaptureSession()
        captureSession.sessionPreset = .photo
        
        guard let backCamera = AVCaptureDevice.default(for: .video) else {
            print("Unable to access back camera!")
            return
        }
        
        do {
            let input = try AVCaptureDeviceInput(device: backCamera)
            
            if captureSession.canAddInput(input) {
                captureSession.addInput(input)
                
                stillImageOutput = AVCapturePhotoOutput()
                
                if captureSession.canAddOutput(stillImageOutput) {
                    captureSession.addOutput(stillImageOutput)
                    
                    videoPreviewLayer = AVCaptureVideoPreviewLayer(session: captureSession)
                    videoPreviewLayer.videoGravity = .resizeAspectFill
                    videoPreviewLayer.connection?.videoOrientation = .portrait
                    
                    view.layer.addSublayer(videoPreviewLayer)
                    
                    captureSession.startRunning()
                }
            }
        } catch let error {
            print("Error unable to initialize back camera: \(error.localizedDescription)")
        }
    }

    func captureOutput(_ output: AVCaptureOutput, didOutput sampleBuffer: CMSampleBuffer, from connection: AVCaptureConnection) {
        // 여권 스캔 로직을 구현합니다.
    }
}

Vision 프레임워크를 사용하여 여권 인식

AVFoundation으로 카메라에 액세스하고 여권을 촬영했다면, Vision 프레임워크를 사용하여 여권의 정보를 추출할 수 있습니다. Vision 프레임워크는 이미지 및 비디오 분석을 위한 강력한 도구를 제공합니다.

import Vision

func processPassport(image: CIImage) {
    let faceDetectionRequest = VNDetectFaceRectanglesRequest(completionHandler: { request, error in
        guard let results = request.results as? [VNFaceObservation] else { return }

        for face in results {
            let boundingBox = face.boundingBox
            // 얼굴 영역을 찾고 여권의 필요한 정보를 추출합니다.
        }
    })

    let imageRequestHandler = VNImageRequestHandler(ciImage: image, orientation: .up, options: [:])

    do {
        try imageRequestHandler.perform([faceDetectionRequest])
    } catch {
        print("Error performing face detection: \(error)")
    }
}

위의 예시 코드는 Vision 프레임워크를 사용하여 여권에서 얼굴을 찾고 해당 영역에서 필요한 정보를 추출하는 방법을 보여줍니다.

여권 스캐너에 다양한 기능과 사용자 경험을 추가할 수 있으며, 위에서 보여준 기본적인 코드를 사용하여 사용자가 쉽게 여권을 스캔할 수 있는 iOS 어플리케이션을 구축할 수 있습니다.

이제 여권 스캐너 어플리케이션을 구축해 보세요!