diff --git a/extras/mmpose/README.md b/extras/mmpose/README.md new file mode 100644 index 0000000..13b2a49 --- /dev/null +++ b/extras/mmpose/README.md @@ -0,0 +1,25 @@ +# Finetuning MMPose models + +See: + +
+ +```bash +docker build --progress=plain -f extras/mmdeploy/dockerfile -t rpt_mmdeploy . + +./extras/mmpose/run_container.sh +``` + +```bash +cd /mmpose/ + +pip3 install "albumentations<1.4" + +python3 ./tools/train.py \ + /mmpose/projects/rtmpose/rtmpose/body_2d_keypoint/rtmpose-m_8xb256-420e_coco-384x288.py \ + --amp \ + --cfg-options \ + load_from=https://download.openmmlab.com/mmpose/v1/projects/rtmposev1/rtmpose-m_simcc-body7_pt-body7_420e-384x288-65e718c4_20230504.pth \ + train_batch_size=16 val_batch_size=16 + +``` diff --git a/extras/mmpose/run_container.sh b/extras/mmpose/run_container.sh new file mode 100755 index 0000000..2b138c8 --- /dev/null +++ b/extras/mmpose/run_container.sh @@ -0,0 +1,11 @@ +#! /bin/bash + +xhost + +docker run --privileged --rm --network host -it \ + --gpus all --shm-size=16g --ulimit memlock=-1 --ulimit stack=67108864 \ + --volume "$(pwd)"/:/RapidPoseTriangulation/ \ + --volume "$(pwd)"/../datasets/coco2017/annotations/:/mmpose/data/coco/annotations/ \ + --volume "$(pwd)"/../datasets/coco2017/images/:/mmpose/data/coco/ \ + --volume /tmp/.X11-unix:/tmp/.X11-unix \ + --env DISPLAY --env QT_X11_NO_MITSHM=1 \ + rpt_mmdeploy