crosstyan/RapidPoseTriangulation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update skelda and dockerfile.

Browse Source
This commit is contained in:
Daniel
2024-08-05 13:14:42 +02:00
parent aea1d8bfb2
commit d935e1cda5
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@ -1,8 +1,8 @@
FROM nvcr.io/nvidia/tensorflow:22.08-tf2-py3 FROM nvcr.io/nvidia/tensorflow:22.08-tf2-py3
ARG DEBIAN_FRONTEND=noninteractive ARG DEBIAN_FRONTEND=noninteractive
ENV LANG C.UTF-8 ENV LANG=C.UTF-8
ENV LC_ALL C.UTF-8 ENV LC_ALL=C.UTF-8
WORKDIR / WORKDIR /
RUN apt-get update && apt-get install -y --no-install-recommends feh RUN apt-get update && apt-get install -y --no-install-recommends feh
@ -22,6 +22,8 @@ RUN pip3 install --upgrade --no-cache-dir "torchvision<0.15"
# Install MMPose # Install MMPose
RUN pip3 install --upgrade --no-cache-dir openmim RUN pip3 install --upgrade --no-cache-dir openmim
RUN mim install mmengine RUN mim install mmengine
RUN mim install "mmcv>=2,<2.2.0"
RUN mim install "mmdet>=3"
RUN mim install "mmpose>=1.1.0" RUN mim install "mmpose>=1.1.0"
# Fix an error when importing mmpose # Fix an error when importing mmpose
RUN pip3 install --upgrade --no-cache-dir numpy scipy RUN pip3 install --upgrade --no-cache-dir numpy scipy

1
scripts/test_skelda_dataset.py
View File

@ -421,7 +421,6 @@ def main():
joint_names_net=joint_names_3d, joint_names_net=joint_names_3d,
joint_names_use=eval_joints, joint_names_use=eval_joints,
save_error_imgs=output_dir, save_error_imgs=output_dir,
pred_imgpaths=all_paths,
) )
_ = evals.pcp.run_eval( _ = evals.pcp.run_eval(
labels, labels,

2
skelda

Submodule skelda updated: 85db8d366d...acae51f7ba