new test torch

Dockerfile

@@ -5,9 +5,10 @@ WORKDIR /code
 #COPY ./requirements.txt /code/requirements.txt
 
 #RUN pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.ht
-RUN pip install torch==1.9.1 torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
-RUN pip install -U openmim
-RUN mim install mmengine
+RUN pip install torch==1.9.1 -f https://download.pytorch.org/whl/torch_stable.ht
+RUN pip install torchvision==0.10.1 -f https://download.pytorch.org/whl/torch_stable.ht
+RUN pip install --no-cache-dir --upgrade -U openmim
+RUN mim install --no-cache-dir --upgrade mmengine
 RUN mim install "mmcv>=2.0.1"
 RUN mim install "mmdet>=3.1.0"
 
@@ -15,15 +16,15 @@ RUN git clone https://github.com/open-mmlab/mmpose.git
 
 # Change directory to mmpose
 WORKDIR /code/mmpose
-RUN pip install -r requirements.txt
-RUN pip install -e .
+RUN pip install --no-cache-dir --upgrade requirements.txt
+RUN pip install --no-cache-dir --upgrade -e .
 
 # How do we change it back with ..
 WORKDIR /..
 
 # RUN mim install "mmpose>=1.1.0"
 
-RUN pip install gradio
+RUN pip install --no-cache-dir --upgrade gradio
 
 #https://stackoverflow.com/questions/55313610/importerror-libgl-so-1-cannot-open-shared-object-file-no-such-file-or-directo
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
@@ -45,4 +46,4 @@ WORKDIR $HOME/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
-CMD ["python", "app.py"]
+CMD ["python", "main.py"]

app.py

@@ -1,30 +0,0 @@
-import mmpose
-print(mmpose.__version__)
-from mmpose.apis import MMPoseInferencer
-inferencer = MMPoseInferencer('human')
-print("[INFO]: Imported modules!!")
-
-import gradio as gr
-
-def greet(photo):
-    print("[INFO]: Downloaded models!")
-    result_generator = inferencer(photo)
-
-    print("[INFO]: Visualizing results!")
-    vis, pred = next(result_generator)
-    return vis
-
-# # specify detection model by alias
-# # the available aliases include 'human', 'hand', 'face', 'animal',
-# # as well as any additional aliases defined in mmdet
-# inferencer = MMPoseInferencer(
-#     # suppose the pose estimator is trained on custom dataset
-#     pose2d='custom_human_pose_estimator.py',
-#     pose2d_weights='custom_human_pose_estimator.pth',
-#     det_model='human'
-# )
-if __name__ == '__main__':
-
-    demo = gr.Interface(fn=greet, inputs=gr.Image(source="webcam"), outputs=gr.Image())
-    
-    demo.launch()