hed-tool

Holistically-Nested Edge Detection(HED, https://arxiv.org/abs/1504.06375)を用いて画像の細線化を行うツールです。

TensorpackのHED実装に基づいてツール化したものです。 HEDの出力をそのまま用いるのではなく、その結果をさらに繊細化する処理を追加しています。実装はpix2pix-tensorflowのprocess.pyをもとにしています。

必要なもの

• TensorFlow
• Tensorpack
• GNU Octave

インストール方法

GNU OctaveとPiotr's Computer Vision Matlab Toolbox他が必要です。

Debian/Ubuntu向けにansible playbookを用意してあります。それ以外の環境については、Qiitaの解説記事あるいはpix2pix-tensorflowのDockerfileを参考にしてください。

HED事前訓練モデル

Tensorpack pretrained modelのページからHED_pretrained_bsds.npzをダウンロードしてください。

実行

以下のように実行してください。

python hed-tool.py --load HED_pretrained_bsds.npz -o output.png input.png

オプション解説

• --load 訓練済みモデル(npzファイル)の指定
• --output / -o 出力ファイル名
• --dir 入力ディレクトリ(ディレクトリ上にあるファイルをまとめて処理する)
• --skip-postprocess / -s Octaveによる細線化処理を行わない

ライセンス

tensorpackのライセンスに従いApache License 2.0とします。