diff --git a/README.md b/README.md index 40e9dc92f3b5fcc87201376b4f6c6c5dc6f2c6ad..da7ec860fbaa21f1f42f915a144196ea10f44878 100644 --- a/README.md +++ b/README.md @@ -6,19 +6,19 @@ Re-implementation of the paper "[*EDeNN: Event Decay Neural Networks for low lat Brings the characteristics of convolutions to event cameras, while acting on the event stream without a loss of information like traditional convolution-bsed networks. -## (Old) Installation [Once the code is finished this will be updated] +## Installation [Once the code is finished this will be updated where needed] 1. Clone this repository ```bash - git clone https://gitlab.surrey.ac.uk/cw0071/edenn.git + git clone https://gitlab.surrey.ac.uk/ah02299/edenn.git cd ./edenn/ ``` 1. Set up conda environment ```bash - conda env create -f environment.yml + conda env create -f docker/environment.yml conda activate edenn pip install -e . ``` @@ -55,7 +55,7 @@ Brings the characteristics of convolutions to event cameras, while acting on the rm ./data/test.tar ``` -## Usage +## Usage (from the old EdeNN code) ### Testing diff --git a/docker/Dockerfile b/docker/Dockerfile index f1e1c1e57f20036f331449470ae3582022befec7..621ce96e17835a44ceb068285b840bcb4ba0f00e 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,14 +1,18 @@ -FROM nvidia/cuda:12.3.1-devel-ubuntu22.04 +# Dockerfile containing the necessary requirements to run the edenn++ repository +# It also needs to have the environment.yml file describing the conda env in the same directory so it can be accessed here. + +FROM nvidia/cuda:11.3.1-devel-ubuntu20.04 MAINTAINER Alejandro Hernández DÃaz -# Update all that stuff to make it niceeee + +ARG DEBIAN_FRONTEND=noninteractive + RUN apt-get update \ - && apt-get install -y build-essential \ + && apt-get install -y build-essential ffmpeg libsm6 libxext6 \ && apt-get install -y wget \ && apt-get clean \ && rm -rf /var/lib/apt/lists/* - # Install miniconda ENV CONDA_DIR /opt/conda RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \