女子プロゴルファーの顔診断AIを作ってみた② - Qiita

1. はじめに 前回までにデータ前処理の部分である画像の収集、顔部分の取得、データの水増しまでを実施しました。 今回は、実際にモデルを作成して、精度を確かめました。 ・自分で構築したモデル ・転移学習(VGG16) の２つについて書こうと思います。 2. 訓練データとテストデータに分けたファイルの作成 from PIL import Image import os, glob import numpy as np from PIL import ImageFile import cv2 from keras.utils.np_utils import to_categorical # IOError: image file is truncated回避のため ImageFile.LOAD_TRUNCATED_IMAGES = True # 訓練データの作成 # それぞれ空のリストを作成 i