Tensorflowで人の顔を見分ける(２)

目的

　Tensorflowで人の顔を分類したい！

参考リンク

　TensorFlowでアニメゆるゆりの制作会社を識別する - kivantium活動日記

動作環境

　Windows7

　Python3.5.3 （過去ログご参考）

やりかた

　全体の流れ

　　１．それぞれの人の顔の画像を複数集める

　　２．画像から人の顔だけを切出す

　　３．切出した顔をチェックして不要なものを削除

　　４．学習用データと検証用データに分ける

　　５．学習する

　　６．学習結果をみる

　　７．見分ける

　この投稿では４項

　　４．学習用データと検証用データに分ける

　　　　　※ランダムにデータを学習用（７割）と検証用（３割）に分ける

　　　　　※学習用データは train.txt へ

　　　　　※検証用データは test.txt へ

　　　　　※各行に、画像データの相対パス、スペース、分類結果を持つ

　　　　　※分類結果は０から番号付け（０なら１人目、１なら２人目）

　　　１）Pythonで以下を実行

# -*- coding: utf-8 -*
import sys
import os
import random

wari  = 0.7
train = open('train.txt', 'w')
test  = open('test.txt', 'w')

files = os.listdir('１人目の画像フォルダ')
for file in files:
  imgfile = '１人目の画像フォルダ' + file
  f = random.random()
  if f > wari:
    test.write(imgfile)
    test.write(" 0\n")
  else:
    train.write(imgfile)
    train.write(" 0\n")

files = os.listdir('２人目の画像フォルダ')
for file in files:
  imgfile = '２人目の画像フォルダ' + file
  f = random.random()
  if f > wari:
    test.write(imgfile)
    test.write(" 1\n")
  else:
    train.write(imgfile)
    train.write(" 1\n")

train.close()
test.close()

Good Luck !!!