アニメの顔を切り抜く

4668 ワード
アニメ OpenCV データセット 機械学習 Python Python テキストリンク

アニメ画像から顔を切り抜く方法です

アニメデータセットをつくりたい

画風変換などで、アニメ画像のデータセットを作るときに、顔だけ切り抜きたい時があります。
大量の画像を自動で処理したい。

animefaceで切り抜ける

OpenCVでanimefaceを使うことで、自動処理できます。

コード

lbpcascade_animefaceをダウンロードします。

!wget https://raw.githubusercontent.com/nagadomi/lbpcascade_animeface/master/lbpcascade_animeface.xml

import cv2

def crop_anime_face(path,write_path):
  face_detector = cv2.CascadeClassifier('lbpcascade_animeface.xml') # initialize detector
  input_img_name = path
  img = cv2.imread(input_img_name)

  faces = face_detector.detectMultiScale(img) # detect
  for x, y, w, h in faces:
    left = x
    top = y
    right = x + w
    bottom = y + h
    face_img = bg[int(top):int(bottom), int(left):int(right)] # crop
    cv2.imwrite(write_path,face_img)

🐣

フリーランスエンジニアです。
お仕事のご相談こちらまで
[email protected]

Core MLやARKitを使ったアプリを作っています。
機械学習／AR関連の情報を発信しています。

Twitter
Medium