diff --git a/docs/challenge.md b/docs/challenge.md index 65f57e450..5c4e36cee 100644 --- a/docs/challenge.md +++ b/docs/challenge.md @@ -61,7 +61,7 @@ Make sure you have [Docker](https://docs.docker.com/engine/install/ubuntu/) with ``` docker system prune -a ``` - *Note 2:* You can modify submission.sh file if your agent needs any custom modifications (e.g. command-line arguments). For example, agent type may be changed by changing `--agent_type` argument in the in the `./scripts/test_local.sh` or `projects/habitat_ovmm/eval_baselines_agent.py`. Otherwise, nothing to do. Default submission.sh is simply a call to agent in `agent.py` + *Note 2:* You can modify `submission.sh` file if your agent needs any custom modifications (e.g. command-line arguments). For example, agent type may be changed by changing `--agent_type` (and corresponding `--baseline_config_path`) argument in the `./scripts/test_local.sh` or `projects/habitat_ovmm/eval_baselines_agent.py`. Otherwise, nothing to do. Default submission.sh is simply a call to agent in `agent.py` 1. Download all the required data into the `home-robot/data` directory (see [Habitat OVMM readme](projects/habitat_ovmm/README.md)). Then in your `docker run` command mount `home-robot/data` data folder to the `home-robot/data` folder in the Docker image (see `./scripts/test_local.sh` for reference). diff --git a/projects/habitat_ovmm/docker/ovmm_baseline.Dockerfile b/projects/habitat_ovmm/docker/ovmm_baseline.Dockerfile index 35f413f0b..ef6859ecf 100644 --- a/projects/habitat_ovmm/docker/ovmm_baseline.Dockerfile +++ b/projects/habitat_ovmm/docker/ovmm_baseline.Dockerfile @@ -11,12 +11,23 @@ RUN /bin/bash -c "\ && pip install -r src/home_robot/home_robot/perception/detection/detic/Detic/requirements.txt \ " -# download pretrained checkpoint +# download pretrained Detic checkpoint RUN mkdir -p home-robot/src/home_robot/home_robot/perception/detection/detic/Detic/models && \ wget https://dl.fbaipublicfiles.com/detic/Detic_LCOCOI21k_CLIP_SwinB_896b32_4x_ft4x_max-size.pth \ -O home-robot/src/home_robot/home_robot/perception/detection/detic/Detic/models/Detic_LCOCOI21k_CLIP_SwinB_896b32_4x_ft4x_max-size.pth \ --no-check-certificate +# download pretrained skills +RUN /bin/bash -c "\ + mkdir -p home-robot/data/checkpoints \ + && cd home-robot/data/checkpoints \ + && wget https://dl.fbaipublicfiles.com/habitat/data/baselines/v1/ovmm_baseline_home_robot_challenge_2023.zip \ + -O ovmm_baseline_home_robot_challenge_2023.zip \ + --no-check-certificate \ + && unzip ovmm_baseline_home_robot_challenge_2023.zip \ + && rm ovmm_baseline_home_robot_challenge_2023.zip \ + " + # add baseline agent code ADD eval_baselines_agent.py /home-robot/projects/habitat_ovmm/agent.py