Skip to content

GitLab

Commit 1fd8e1a1 authored by Yizhou Wang's avatar Yizhou Wang
Browse files

Update README.md

parent 1d3dead7
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 6 deletions
README.md
View file @ 1fd8e1a1
...@@ -10,7 +10,7 @@ conda activate rodnet ...@@ -10,7 +10,7 @@ conda activate rodnet
Install pytorch. Install pytorch.
``` ```
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
``` ```
Setup RODNet package. Setup RODNet package.
...@@ -36,11 +36,6 @@ python tools/train.py --config configs/<CONFIG_FILE> \ ...@@ -36,11 +36,6 @@ python tools/train.py --config configs/<CONFIG_FILE> \
--log_dir checkpoints/ --log_dir checkpoints/
``` ```
### Pretrained Models
Please download the pretrained models from the list below:
- [rodnet_cdc_win16_cruw_mini_epoch50.pkl](https://drive.google.com/file/d/19j-jy5wVOwYjUS5SPvoF-MjA06PepYnA/view?usp=sharing)
## Inference ## Inference
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment