The latest offiicial tensorboard uses name tensoboard, the tensorboar…

…d-pytorch in renamed to tensorboardX.

README.md

@@ -46,7 +46,7 @@ Additional features not mentioned in the [report](https://arxiv.org/pdf/1702.021
 ### Prerequisites
   - A basic pytorch installation. The code follows **0.2**. If you are using old **0.1.12**, you can checkout 0.1.12 branch.
   - Python packages you might not have: `cffi`, `opencv-python`, `easydict` (similar to [py-faster-rcnn](https://github.com/rbgirshick/py-faster-rcnn)). For `easydict` make sure you have the right version. I use 1.6.
-  - [tensorboard-pytorch](https://github.com/lanpa/tensorboard-pytorch) to visualize the training and validation curve.
+  - [tensorboard-pytorch](https://github.com/lanpa/tensorboard-pytorch) to visualize the training and validation curve. Please build from source to use the latest tensorflow-tensorboard.
   - ~~Docker users: Since the recent upgrade, the docker image on docker hub (https://hub.docker.com/r/mbuckler/tf-faster-rcnn-deps/) is no longer valid. However, you can still build your own image by using dockerfile located at `docker` folder (cuda 8 version, as it is required by Tensorflow r1.0.) And make sure following Tensorflow installation to install and use nvidia-docker[https://github.com/NVIDIA/nvidia-docker]. Last, after launching the container, you have to build the Cython modules within the running container.~~
 
 ### Installation

lib/model/train_val.py

@@ -7,7 +7,7 @@
 from __future__ import division
 from __future__ import print_function
 
-import tensorboard as tb
+import tensorboardX as tb
 
 from model.config import cfg
 import roi_data_layer.roidb as rdl_roidb

lib/nets/network.py

@@ -7,8 +7,6 @@
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow as tf
-
 import math
 
 import torch
@@ -29,7 +27,7 @@
 
 from model.config import cfg
 
-import tensorboard as tb
+import tensorboardX as tb
 
 class Network(nn.Module):
   def __init__(self, batch_size=1):