xmrt commited on
Commit
7cd5862
·
1 Parent(s): 0ab69bb

new test mmdet

Browse files
Files changed (2) hide show
  1. Dockerfile +5 -5
  2. app.py +4 -1
Dockerfile CHANGED
@@ -5,9 +5,9 @@ WORKDIR /code
5
  #COPY ./requirements.txt /code/requirements.txt
6
 
7
  #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
8
- RUN pip install --no-cache-dir torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
9
- RUN pip install --no-cache-dir -U openmim
10
- RUN mim install --no-cache-dir mmengine
11
  RUN mim install "mmcv>=2.0.1"
12
  RUN mim install "mmdet>=3.1.0"
13
 
@@ -15,7 +15,7 @@ RUN git clone https://github.com/open-mmlab/mmpose.git
15
 
16
  # Change directory to mmpose
17
  WORKDIR /code/mmpose
18
- RUN pip install --no-cache-dir -r requirements.txt
19
  RUN pip install -e .
20
 
21
  # How do we change it back with ..
@@ -23,7 +23,7 @@ WORKDIR /..
23
 
24
  # RUN mim install "mmpose>=1.1.0"
25
 
26
- RUN pip install --no-cache-dir gradio
27
 
28
  #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
29
  RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
 
5
  #COPY ./requirements.txt /code/requirements.txt
6
 
7
  #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
8
+ RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
9
+ RUN pip install -U openmim
10
+ RUN mim install mmengine
11
  RUN mim install "mmcv>=2.0.1"
12
  RUN mim install "mmdet>=3.1.0"
13
 
 
15
 
16
  # Change directory to mmpose
17
  WORKDIR /code/mmpose
18
+ RUN pip install -r requirements.txt
19
  RUN pip install -e .
20
 
21
  # How do we change it back with ..
 
23
 
24
  # RUN mim install "mmpose>=1.1.0"
25
 
26
+ RUN pip install gradio
27
 
28
  #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
29
  RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
app.py CHANGED
@@ -2,12 +2,15 @@ import mmpose
2
  print(mmpose.__version__)
3
  from mmpose.apis import MMPoseInferencer
4
  inferencer = MMPoseInferencer('human')
 
5
 
6
  import gradio as gr
7
 
8
  def greet(photo):
9
-
10
  result_generator = inferencer(photo)
 
 
11
  vis, pred = next(result_generator)
12
  return vis
13
 
 
2
  print(mmpose.__version__)
3
  from mmpose.apis import MMPoseInferencer
4
  inferencer = MMPoseInferencer('human')
5
+ print("[INFO]: Imported modules!!")
6
 
7
  import gradio as gr
8
 
9
  def greet(photo):
10
+ print("[INFO]: Downloaded models!")
11
  result_generator = inferencer(photo)
12
+
13
+ print("[INFO]: Visualizing results!")
14
  vis, pred = next(result_generator)
15
  return vis
16