MXNetでTensorBoardを使う

MXNet

github.com

環境

Windows 10 Pro
GPUなし
Python 3.6.6（venv使用）

バージョンの確認（pip freeze）

absl-py==0.5.0
astor==0.7.1
astroid==2.0.4
certifi==2018.8.24
chardet==3.0.4
colorama==0.4.0
cycler==0.10.0
gast==0.2.0
gluoncv==0.3.0b20181010
graphviz==0.8.4
grpcio==1.15.0
h5py==2.8.0
idna==2.6
isort==4.3.4
Keras-Applications==1.0.6
Keras-Preprocessing==1.0.5
kiwisolver==1.0.1
lazy-object-proxy==1.3.1
Markdown==3.0.1
matplotlib==3.0.0
mccabe==0.6.1
mxboard==0.1.0
mxnet==1.3.0
numpy==1.14.6
Pillow==5.3.0
protobuf==3.6.1
pylint==2.1.1
pyparsing==2.2.2
python-dateutil==2.7.3
requests==2.18.4
scipy==1.1.0
six==1.11.0
tensorboard==1.11.0
tensorflow==1.11.0
termcolor==1.1.0
tqdm==4.26.0
typed-ast==1.1.0
urllib3==1.22
Werkzeug==0.14.1
wrapt==1.10.11

サンプルコード

import mxnet as mx
from gluoncv import model_zoo
from matplotlib import pyplot as plt

from mxboard import SummaryWriter

net = model_zoo.get_model('yolo3_darknet53_voc', pretrained=True)

a = mx.nd.ones((1,3,512,512))
net.hybridize()
ids, scores, bb = net(a)

with SummaryWriter(logdir='./logs') as sw:
    sw.add_graph(net)

ブラウザで開く

tensorboard --logdir=./logs --host=127.0.0.1 --port=8888
  • Edgeではうまく開けなかった
  • FireFoxで問題なく開けた

最後に

version0.1.0のまま5ヶ月更新されていないのが気になる