前回はAzure上でKerasがとりあえず動くところまで確認出来たので、今回は実際にKerasでFCNを動かしてみます。対象とするFCNはWindows環境で動作を確認したkeras-fcnを試してみます。
まず、PCからSSH等でAzureに接続してVOC2012のデータセットを取得します。
$ wget "http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar"
$ tar -xvz voc2012/VOCtrainval_11-May-2012.tar
$ mkdir ~/.keras/datasets
$ mv VOC2012 ~/Datasets
次にkeras-fcnをクローンして必要なライブラリをインストールします。
$ mkdir src
$ cd src
$ git clone https://github.com/JihongJu/keras-fcn.git
$ cd keras-fcn
$ pip install --editable .
$ cd voc2012
init_args.ymlを下記のようにVOC2012への相対パスに変更します。
image_set_loader:
train:
image_set: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/ImageSets/Segmentation/train.txt
image_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/JPEGImages/
label_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/SegmentationClass
target_size:
- 224
- 224
image_format: 'jpg'
color_mode: 'rgb'
label_format: 'png'
val:
image_set: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/ImageSets/Segmentation/val.txt
image_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/JPEGImages/
label_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/SegmentationClass
target_size:
- 224
- 224
image_format: 'jpg'
color_mode: 'rgb'
label_format: 'png'
test:
image_set: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/ImageSets/Segmentation/test.txt
image_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/JPEGImages/
label_dir: ../../../.keras/datasets/VOC2012/VOCdevkit/VOC2012/SegmentationClass
target_size:
- 224
- 224
image_format: 'jpg'
color_mode: 'rgb'
label_format: 'png'
下記にて学習を実行します。
$ python train.py