GANでハンコジェネレーターを作ってみた #Python

はじめに

河野さんが行政改革で着手した「ハンコ廃止」

それに対して「本人確認の最も有効な手段」と、異を唱えるハンコ議連

（参考「はんこ議連「押印は本人確認に有効」廃止に対応要請」）

よろしい･･･では敢えて問おう！

「ハンコって本当に本人確認能力あるのか？」

この逆説を検証するために、GAN（Generative Adversarial Network）で　“ハンコジェネレーター”を作ってみた。

※結論だけ見たいせっかちさんはコチラ　⇒　結論

準備

サンプルデータ作成

とりあえず、力業でハンコのサンプルを作ってみた。（全国の石原さん。あしからず・・・）
- A4用紙を2cm×2cmの格子状に区切って、縦14×横10の計140個のハンコを押印
- ひたすら押印！押印！押印！押印！・・・・・・・・・・・・（なんか、とっても偉い人の苦労が分かった気がする）
- スキャン→画像処理ソフトで格子線を消して余白をトリミング
- できた！（下図）

なんか、こう見ると欠けていたり、傾いていたり、薄かったり、同じ三文判を同じ人が押しても、結構いろんなバリエーションが出来る。

次に、ハンコ１つ１つを分割する

画像処理ソフトで手作業は大変だし、MNISTの64*64pixelにデータ仕様に合わせたいので、簡単なプログラムを作った。（参考）

import os
import glob
import numpy as np
import skimage.util
import skimage.io
import skimage.transform
out_size = 64 # 最終出力サイズ（横、縦）
img = skimage.io.imread('data/low/hanko.jpg')
print(img.shape)
# (3307, 2340, 3)
#ここで割り切れない中途半端なサイズだとエラーが出るので、リサイズ
img_ = skimage.transform.resize(img, (64*14, 64*10), anti_aliasing=True)
img_gray = skimage.color.rgb2gray(img_) #グレースケール変換
print(img_gray.shape)
#(3307, 2340)
blocks = skimage.util.view_as_blocks(img_gray, (img_gray.shape[0]//14, img_gray.shape[1]//10)).squeeze()
print(blocks.shape)
# (14, 10, 64, 64)
path_to_out = 'data/low/hanko/' #出力先
os.makedirs(path_to_out, exist_ok=True) #無ければ作成
file_list = glob.glob(path_to_out + "hanko*jpg") #存在チェック＆削除
for f in file_list: os.remove(f)
idx = 0
for i in range(14):
    for j in range(10):
        skimage.io.imsave(path_to_out + 'hanko{:0=3}.jpg'.format(idx), blocks[i,j])
        idx += 1

GANの実装

プログラムは下記書籍で紹介されているものを基本そのまま流用。（中身の詳しい原理などは、ここでは割愛）
- 「つくりながら学ぶ！PyTorchによる発展ディープラーニング」
- 第5章 GANによる画像生成（DCGAN, Self-Attention GAN）
- 書籍：https://www.amazon.co.jp/dp/4839970254
- サンプルプログラム：https://github.com/YutaroOgawa/pytorch_advanced/tree/master/3_semantic_segmentation
一部、下記のように変更
- フォルダパスを適当に書き換えて、データサンプル数も調整。
- もともとのMNIST（手書き数字データセット）の場合200枚だったけど、今回は60枚も少ない140枚。大丈夫かな･･･

  def make_datapath_list():
    train_img_list = list()
    img_num = 140 #サンプル数
    for img_idx in range(img_num):
        img_path = "../data/low/hanko/hanko{:0=3}.jpg".format(img_idx)
        train_img_list.append(img_path)
    return train_img_list

試行錯誤も想定して、epoch数をコマンド引数から取れるように、学習実行部分をちょこっとだけ改造

  import sys
  args = sys.argv
  num_epochs = int(args[1]) if len(args)>1 else 300 
  G_update, D_update = train_model(
      G, D, dataloader=train_dataloader, num_epochs=num_epochs)

学習＆実行

さあ、準備は整った！やってみよう。
今回は、GANの典型である「DCGAN」と、その改良版である「SAGAN」の2種類のプログラムを試した。
（参考）実行環境
- CPU : Intel(R) Xeon(R) CPU E3-1225 v6 @ 3.30GHz
- OS : Windows 10 Pro for Workstations
- GPU : NVIDIA Quadro P4000