Skip to content

Commit cfefccd

Browse files
adityaarun1ruotianluo
authored andcommitted
Update README
1 parent 753667c commit cfefccd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ Additional features not mentioned in the [report](https://arxiv.org/pdf/1702.021
5959

6060
### Prerequisites
6161
- A basic pytorch installation. The code follows **1.0**. If you are using old **0.1.12** or **0.2** or **0.3** or **0.4**, you can checkout the corresponding branch.
62-
- Torchvision 0.3. This code uses `torchvision.ops` for `nms`, `roi_pool` and `roi_align`
62+
- Torchvision **0.3**. This code uses `torchvision.ops` for `nms`, `roi_pool` and `roi_align`
6363
- Python packages you might not have: `opencv-python`, `easydict` (similar to [py-faster-rcnn](https://github.com/rbgirshick/py-faster-rcnn)). For `easydict` make sure you have the right version. Xinlei uses 1.6.
6464
- [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.
6565
- ~~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.~~

0 commit comments

Comments
 (0)