From 675445afaa35aa27da6344e7c19e2d7f1cdee9a8 Mon Sep 17 00:00:00 2001
From: WEI CHEN <77597327+Wei-Chen-hub@users.noreply.github.com>
Date: Wed, 18 Oct 2023 12:43:19 +0800
Subject: [PATCH] Update inference docker description

---
 README.md | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/README.md b/README.md
index 5c524bd..ba345ea 100644
--- a/README.md
+++ b/README.md
@@ -213,6 +213,17 @@ sh slurm_test.sh {JOB_NAME} {NUM_GPU} {TRAIN_OUTPUT_DIR} {CKPT_ID}
   - We are working on that, please stay tuned!
     Currently, this repo supports SMPL-X estimation and a simple visualization (overlay of SMPL-X vertices).
 
+## Docker Support (Early Stage)
+```
+docker pull wcwcw/smplerx_inference:v0.2
+docker run  --gpus all -v <vid_input_folder>:/smplerx_inference/vid_input \
+        -v <vid_output_folder>:/smplerx_inference/vid_output \
+        wcwcw/smplerx_inference:v0.2 --vid <video_name>.mp4
+# Currently any customization need to be applied to /smplerx_inference/smplerx/inference_docker.py
+```
+- We recently developed a docker for inference at docker hub.
+- This docker image uses SMPLer-X-H32 as inference baseline and was tested at RTX3090 & WSL2.
+
 ## References
 - [Hand4Whole](https://github.com/mks0601/Hand4Whole_RELEASE)
 - [OSX](https://github.com/IDEA-Research/OSX)