NeLy-EPFL / nmf2-paper

Code for generating some figures in for the NeuroMechFly v2 paper.

NeuroMechFly v2 Manuscript Repository

This repository contains the scripts to generate the figures and videos for the NeuroMechFly v2 paper. It is for archival purposes and will not be maintained.

To use NeuroMechFly, see the FlyGym repository and visit neuromechfly.org.

Instructions

Install the resubmission-20240501 snapshot of FlyGym.

Clone this repository:

git clone [e-mail]:NeLy-EPFL/nmf2-paper.git
cd nmf2-paper
git checkout as-published

Create the conda environment using the provided environment.yml file:

conda env create --file environment.yml

If you are working on a machine without a display (e.g., a server), you will need to switch the renderer to EGL:

conda activate nmf2-paper
conda env config vars set MUJOCO_GL=egl PYOPENGL_PLATFORM=egl
conda deactivate

Some tasks require time-consuming data collection and training. To save time, you can download the data and trained models:

./download_data.sh

Generate the figures and videos by running the following script:

conda activate nmf2-paper
./generate_figures.sh

Note that the results may vary slightly due to differences in system architecture.