From d935e1cda53cd62644472daa9be06090f48ad1a4 Mon Sep 17 00:00:00 2001 From: Daniel Date: Mon, 5 Aug 2024 13:14:42 +0200 Subject: [PATCH] Update skelda and dockerfile. --- Dockerfile | 6 ++++-- scripts/test_skelda_dataset.py | 1 - skelda | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/Dockerfile b/Dockerfile index 827b003..2412887 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,8 +1,8 @@ FROM nvcr.io/nvidia/tensorflow:22.08-tf2-py3 ARG DEBIAN_FRONTEND=noninteractive -ENV LANG C.UTF-8 -ENV LC_ALL C.UTF-8 +ENV LANG=C.UTF-8 +ENV LC_ALL=C.UTF-8 WORKDIR / 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 RUN pip3 install --upgrade --no-cache-dir openmim RUN mim install mmengine +RUN mim install "mmcv>=2,<2.2.0" +RUN mim install "mmdet>=3" RUN mim install "mmpose>=1.1.0" # Fix an error when importing mmpose RUN pip3 install --upgrade --no-cache-dir numpy scipy diff --git a/scripts/test_skelda_dataset.py b/scripts/test_skelda_dataset.py index 153d220..e69092c 100644 --- a/scripts/test_skelda_dataset.py +++ b/scripts/test_skelda_dataset.py @@ -421,7 +421,6 @@ def main(): joint_names_net=joint_names_3d, joint_names_use=eval_joints, save_error_imgs=output_dir, - pred_imgpaths=all_paths, ) _ = evals.pcp.run_eval( labels, diff --git a/skelda b/skelda index 85db8d3..acae51f 160000 --- a/skelda +++ b/skelda @@ -1 +1 @@ -Subproject commit 85db8d366dea4bf772b74dd97fc7b90881f1ba1d +Subproject commit acae51f7ba0e208cbd177230305bb3b022c90106