diff --git a/README.md b/README.md index bbbf564..9a68d53 100644 --- a/README.md +++ b/README.md @@ -31,17 +31,11 @@ git clone https://github.com/mlpc-ucsd/LETR.git ### Step2: Environment Installation ```bash -mkdir -p data -mkdir -p evaluation/data -mkdir -p exp +mkdir -p data evaluation/data exp - -conda create -n letr python anaconda -conda activate letr -conda install -c pytorch pytorch torchvision -conda install cython scipy +conda env create -f environment.yml pip install -U 'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI' -pip install docopt +conda activate letr ``` ### Step3: Data Preparation diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..14a1f51 --- /dev/null +++ b/environment.yml @@ -0,0 +1,14 @@ +name: letr +dependencies: + - numpy + - pytorch==1.8.0 + - torchvision==0.9.0 + - scipy + - opencv + - matplotlib + - jupyter + - nb_conda_kernels + - pip + - pip: + - docopt +