Jiading Fang commited on
Commit
c1e8cfb
·
1 Parent(s): 2512c83

update Dockerfile for local data

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -3
Dockerfile CHANGED
@@ -46,7 +46,7 @@ RUN pip install \
46
  boto3==1.17.83 \
47
  termcolor==1.1.0 \
48
  pyyaml==5.4.1 \
49
- wandb==0.10.31 \
50
  opencv-python==4.5.2.52 \
51
  flow_vis==0.1 \
52
  matplotlib==3.3.4 \
@@ -107,10 +107,15 @@ WORKDIR /workspace/${PROJECT}
107
  # Copy project to workspace folder
108
  COPY . /workspace/${PROJECT}
109
 
 
 
 
 
 
 
110
  # Set environment variables
111
  ENV PYTHONPATH="/workspace/${PROJECT}:$PYTHONPATH"
112
  ENV PYTHONPATH="/workspace/${PROJECT}/externals/camviz:$PYTHONPATH"
113
 
114
  # command
115
- CMD ["python3", "app.py"]
116
- # CMD ["gradio", "app.py"] # debug
 
46
  boto3==1.17.83 \
47
  termcolor==1.1.0 \
48
  pyyaml==5.4.1 \
49
+ wandb==0.14.0 \
50
  opencv-python==4.5.2.52 \
51
  flow_vis==0.1 \
52
  matplotlib==3.3.4 \
 
107
  # Copy project to workspace folder
108
  COPY . /workspace/${PROJECT}
109
 
110
+ # Create data directory
111
+ WORKDIR /data/vidar
112
+ RUN ln -s /workspace/${PROJECT}/models/ /data/vidar/models
113
+ RUN ln -s /workspace/${PROJECT}/scannet_processed /data/vidar/scannet_processed
114
+ WORKDIR /workspace/${PROJECT}
115
+
116
  # Set environment variables
117
  ENV PYTHONPATH="/workspace/${PROJECT}:$PYTHONPATH"
118
  ENV PYTHONPATH="/workspace/${PROJECT}/externals/camviz:$PYTHONPATH"
119
 
120
  # command
121
+ CMD ["python", "app.py"]