fix rgb

Tau-J · Sep 19, 2023 · 9480252 · 9480252
1 parent ae9038a
commit 9480252
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -61,7 +61,7 @@ Run `webui.py`:
 python webui.py
 ```
 
-![image](https://github.com/Tau-J/rtmlib/assets/13503330/eb3913f3-2789-4262-8353-b5e7864aa441)
+![image](https://github.com/Tau-J/rtmlib/assets/13503330/d973888d-d741-49ae-8f46-889fa89f28a7)
 
 Here is also a simple demo to show how to use rtmlib to conduct pose estimation on a single image.
 

diff --git a/webui.py b/webui.py
@@ -43,7 +43,7 @@ def predict(img,
                              scores,
                              openpose_skeleton=openpose_skeleton,
                              kpt_thr=0.4)
-    return img_show
+    return img_show[:, :, ::-1]
 
 
 with gr.Blocks() as demo: