hed-tool
以前pix2pix付属のツールで画像のエッジ化を行う記事を書きましたが、古いバージョンをベースに改造されたCaffeを用いるという点に不満がありました。
他の実装を探してみたところ、TensorpackのサンプルにHED実装と事前訓練モデルが用意されていたので、これをベースにツール化をしました。
インストール
詳細はドキュメントに記載しています。箇条書きで以下にまとめます。
- TensorFlow, Tensorpackのインストール
- GNU OctaveとPiotr's Computer Vision Matlab Toolbox等のインストール
- HED事前訓練モデルのダウンロード
実行例
$ python hed-tool.py --load HED_pretrained_bsds.npz -o output.png input.png
入力画像
ソース: pixabay