diff --git a/README.md b/README.md index 5591619fee77ca338ff81834ac283707cdbf6d94..ef62c54cd0ad5e333a0d0d74192060401112d530 100644 --- a/README.md +++ b/README.md @@ -11,7 +11,7 @@ Brings the characteristics of convolutions to event cameras, while acting on the 1. Clone this repository ```bash - git clone https://github.com/celynw/edenn.git + git clone https://gitlab.surrey.ac.uk/cw0071/edenn.git cd ./edenn/ ``` @@ -20,16 +20,14 @@ Brings the characteristics of convolutions to event cameras, while acting on the ```bash conda env create -f environment.yml conda activate edenn - pip install -e ./edenn/ + pip install -e . ``` For reproducing the other benchmarks, install SLAYER manually: ```bash - # cd submodules/slayerPytorch/ - cd slayerPytorch/ - python setup.py install - cd - + # pip install submodules/slayerPytorch/ + pip install ./slayerpytorch/ ``` 1. Download dataset(s) @@ -39,21 +37,21 @@ Brings the characteristics of convolutions to event cameras, while acting on the wget "http://rpg.ifi.uzh.ch/data/snn_angular_velocity/dataset/train.tar.zst" -O ./data/train.tar.zst zstd -vd ./data/train.tar.zst rm ./data/train.tar.zst - tar -xvf ./data/train.tar + tar -xvf ./data/train.tar -C ./data/ rm ./data/train.tar # Validation set (2.7 GB download, 7.7 GB extracted) wget "http://rpg.ifi.uzh.ch/data/snn_angular_velocity/dataset/val.tar.zst" -O ./data/val.tar.zst zstd -vd ./data/val.tar.zst rm ./data/val.tar.zst - tar -xvf ./data/val.tar + tar -xvf ./data/val.tar -C ./data/ rm ./data/val.tar # Test set (2.6 GB download, 7.4 GB extracted) wget "http://rpg.ifi.uzh.ch/data/snn_angular_velocity/dataset/test.tar.zst" -O ./data/test.tar.zst zstd -vd ./data/test.tar.zst rm ./data/test.tar.zst - tar -xvf ./data/test.tar + tar -xvf ./data/test.tar -C ./data/ rm ./data/test.tar ``` @@ -97,7 +95,7 @@ wget "http://rpg.ifi.uzh.ch/data/snn_angular_velocity/models/pretrained.pt" -O c Then run *this* test script: ```bash -./test_snn.py SNN_baseline cnn-avgp-fc1.pt path-to-dataset/ +./test_snn.py SNN_baseline cnn5-avgp-fc1.pt data/ --nolog ``` ```none diff --git a/data/.gitkeep b/data/.gitkeep new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/environment.yml b/environment.yml index 9bb442218bc6d19da87fa921b8d1251265de3b84..96ecba866a3af552a821b6cadef0bd798d1f5e7b 100644 --- a/environment.yml +++ b/environment.yml @@ -326,5 +326,3 @@ dependencies: - zipp=3.15.0=pyhd8ed1ab_0 - zlib=1.2.13=h166bdaf_4 - zstd=1.5.2=h3eb15da_6 - - pip: - - slayercuda==0.0.0