new test mmdet
Browse files- Dockerfile +5 -5
- 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
|
9 |
-
RUN pip install
|
10 |
-
RUN mim install
|
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
|
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
|
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 |
|