Skip to content
Snippets Groups Projects
Commit 044f7d32 authored by Walters, Celyn Dr (Comp Sci & Elec Eng)'s avatar Walters, Celyn Dr (Comp Sci & Elec Eng)
Browse files

Fix things for end-user run through

parent fed290ce
Branches master
No related tags found
No related merge requests found
......@@ -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
......
......@@ -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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment