Skip to content

Commit 9b0d397

Browse files
committed
Minor updates to full and lite
1 parent 284c697 commit 9b0d397

File tree

3 files changed

+14
-1
lines changed

3 files changed

+14
-1
lines changed

sapiens/README.md

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,17 @@ Folder lite contains docker that builds environment only for the sapiens inferen
2222

2323
Folder full contains Dockerfile that builds environment for the training (fine-tuning) env for sapiens.
2424

25+
Build full version by running:
26+
```
27+
cd full
28+
docker build -t sapiens_img:full .
29+
```
30+
31+
Run container with:
32+
```
33+
./first_run.sh
34+
```
35+
2536
### How to install `mmcv`?
2637

2738
Instructions how to properly build mmcv are [here](https://mmcv.readthedocs.io/en/latest/get_started/installation.html).

sapiens/full/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,14 @@ ENV NVIDIA_DRIVER_CAPABILITIES graphics,utility,compute
99
ENV TZ=Europe/Zagreb
1010
ENV ROSCONSOLE_FORMAT '[${severity}] [${time}] [${node}]: ${message}'
1111

12+
# Install apt packages
1213
RUN apt-get update -y && \
1314
apt-get install -y --no-install-recommends \
1415
build-essential \
1516
wget \
1617
git \
1718
feh \
19+
nano \
1820
ffmpeg \
1921
libsm6 \
2022
libxext6

sapiens/lite/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ RUN conda init bash \
3535
&& conda create --name sapiens_lite python=3.10 \
3636
&& conda activate sapiens_lite \
3737
&& conda install torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia \
38-
&& pip install opencv-python tqdm json-tricks
38+
&& pip install opencv-python tqdm json-tricks mmdet
3939

4040
# Sapiens lite installation
4141
RUN cd $SAPIENS

0 commit comments

Comments
 (0)