Improved doc

BenJamesbabala · Apr 28, 2017 · 670c5d2 · 670c5d2
1 parent ac4607a
commit 670c5d2
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 7 deletions.
diff --git a/doc/demo_overview.md b/doc/demo_overview.md
@@ -28,6 +28,15 @@ The visual GUI should show the original image with the poses blended on it, simi
     <img src="media/shake.gif", width="720">
 </p>
 
+If you choose to visualize a body part or a PAF (Part Affinity Field) heat map, the result should be similar to the following images:
+<p align="center">
+    <img src="media/body_heat_maps.gif", width="720">
+</p>
+
+<p align="center">
+    <img src="media/paf_heat_maps.gif", width="720">
+</p>
+
 
 
 ## Other Important Options

diff --git a/doc/media/body_heat_maps.png b/doc/media/body_heat_maps.png
diff --git a/doc/media/paf_heat_maps.jpg b/doc/media/paf_heat_maps.jpg
diff --git a/doc/media/pose_face_hands.gif b/doc/media/pose_face_hands.gif
diff --git a/src/openpose/experimental/face/faceRenderGpu.cu b/src/openpose/experimental/face/faceRenderGpu.cu
@@ -47,12 +47,9 @@ namespace op
         {
             if (numberFaces > 0)
             {
-                //framePtr      =   width * height * 3
-                //facePtr       =   3 (x,y,score) * #Hand parts * # face
-                const auto threshold = 0.4f;
+                const auto threshold = 0.5f;
                 dim3 threadsPerBlock = dim3{THREADS_PER_BLOCK_1D, THREADS_PER_BLOCK_1D};
                 dim3 numBlocks = dim3{getNumberCudaBlocks(frameSize.width, threadsPerBlock.x), getNumberCudaBlocks(frameSize.height, threadsPerBlock.y)};
-                // log("numberFaces: " + std::to_string(numberFaces), Priority::Low, __LINE__, __FUNCTION__, __FILE__);
                 renderFaceParts<<<threadsPerBlock, numBlocks>>>(framePtr, frameSize.width, frameSize.height, facePtr, numberFaces, threshold, alphaColorToAdd);
                 cudaCheck(__LINE__, __FUNCTION__, __FILE__);
             }

diff --git a/src/openpose/experimental/hand/handRenderGpu.cu b/src/openpose/experimental/hand/handRenderGpu.cu
@@ -70,12 +70,9 @@ namespace op
         {
             if (numberHands > 0)
             {
-                //framePtr      =   width * height * 3
-                //handsPtr =   3 (x,y,score) * #Hand parts * # hands
                 const auto threshold = 0.05f;
                 dim3 threadsPerBlock = dim3{THREADS_PER_BLOCK_1D, THREADS_PER_BLOCK_1D};
                 dim3 numBlocks = dim3{getNumberCudaBlocks(frameSize.width, threadsPerBlock.x), getNumberCudaBlocks(frameSize.height, threadsPerBlock.y)};
-                // log("numberHands: " + std::to_string(numberHands), Priority::Low, __LINE__, __FUNCTION__, __FILE__);
                 renderHandsParts<<<threadsPerBlock, numBlocks>>>(framePtr, frameSize.width, frameSize.height, handsPtr, numberHands, threshold, alphaColorToAdd);
                 cudaCheck(__LINE__, __FUNCTION__, __FILE__);
             }