質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

88.91%

DeepLabV3を使って写真の背景を抽出し白く塗り潰したい

受付中

回答 0

投稿

  • 評価
  • クリップ 0
  • VIEW 168

yamaji1108

score 19

前提・実現したいこと

取り込んだ写真の背景を白くしたいです。
下記が目標です。(photoroomというアプリを使用しました)
イメージ説明

イメージ説明

色々調べたところ、Core MLモデルのDeeplabV3を使用して背景透過処理をされている方の記事を発見しました。
https://tech.basicinc.jp/articles/206

この方はmacOSのアプリのために使っていたため、私の方ではNSImageの部分をUIImageに書き換えました。
また、UIImageをCVPixelBufferに変換する関数(Buffer)は
https://stackoverrun.com/ja/q/12215731
の記事を参考にしました。

自分なりにコードを書いてみましたが、コメントアウト部分の背景を白くする記述が分からない状況です。

該当のソースコード

//略

private func backgroundToWhite(brightnessValue: Float, contrastValue: Float, saturationValue: Float, isUpdatedSegmentation: Bool = false) {
        guard let img = sourceImageView.image?.cropping(to: selectedRect!) else {
            assertionFailure()
            return
        }
        //cropImageViewにimgを格納しておく(比較するため)
        cropImageView.image = img
        //写真サイズの表記をトリミングされたimgのサイズに変更
        imageSizeLabel.text = "\(Int(img.size.width)) x \(Int(img.size.height))"

        //ここから本処理です。
        let image = buffer(from:img)

        let model = DeepLabV3()
        guard let output = try? model.prediction(image: image!) else {
            fatalError("Error")
        }

        for y in 0..<Int(img.size.height) {
            for x in 0..<Int(img.size.width) {
                let getY = Int(img.size.width) - y - 1
                let index: [NSNumber] = [NSNumber(value: getY), NSNumber(value: x)]
                let flag = output.semanticPredictions[index]
                if flag == 0 {
                    // 背景部分を白く変える処理を書きたいです。

                }
            }
        }
        //newImage.addRepresentation(rep)

    }

    private func buffer(from image: UIImage) -> CVPixelBuffer? {
        let attrs = [kCVPixelBufferCGImageCompatibilityKey: kCFBooleanTrue, kCVPixelBufferCGBitmapContextCompatibilityKey: kCFBooleanTrue] as CFDictionary
        var pixelBuffer : CVPixelBuffer?
        let status = CVPixelBufferCreate(kCFAllocatorDefault, Int(image.size.width), Int(image.size.height), kCVPixelFormatType_32ARGB, attrs, &pixelBuffer)
        guard (status == kCVReturnSuccess) else {
        return nil
        }

        CVPixelBufferLockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))
        let pixelData = CVPixelBufferGetBaseAddress(pixelBuffer!)

        let rgbColorSpace = CGColorSpaceCreateDeviceRGB()
        let context = CGContext(data: pixelData, width: Int(image.size.width), height: Int(image.size.height), bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(pixelBuffer!), space: rgbColorSpace, bitmapInfo: CGImageAlphaInfo.noneSkipFirst.rawValue)

        context?.translateBy(x: 0, y: image.size.height)
        context?.scaleBy(x: 1.0, y: -1.0)

        UIGraphicsPushContext(context!)
        image.draw(in: CGRect(x: 0, y: 0, width: image.size.width, height: image.size.height))
        UIGraphicsPopContext()
        CVPixelBufferUnlockBaseAddress(pixelBuffer!, CVPixelBufferLockFlags(rawValue: 0))

        return pixelBuffer
    }
//略

試したこと

DeepLabV3の使い方等を調べましたが、今回私がxcode上で実装したい機能についての記事が上記しか見つかりませんでした。。

理解度が足りていないのは重々承知しております。
関数のヒントやDeepLabV3の関連記事などご教授いただければ、嬉しいです。

よろしくお願いいたします。

補足情報(FW/ツールのバージョンなど)

Xcode Version 11.4.1
Swift Version 5.0.1

  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 過去に投稿した質問と同じ内容の質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

まだ回答がついていません

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 88.91%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

関連した質問

同じタグがついた質問を見る