Coco JSON extractor now working any coco folder

Author: gineshidalgo99

doc/release_notes.md

@@ -85,13 +85,17 @@ OpenPose Library - Release Notes
 
 ## Current version (future OpenPose 1.0.2)
 1. Main improvements:
-    1. Caffe turned into DLL library.
-    2. OpenPose is now completely portable across Windows 10 computers (with Nvidia graphic card).
-    3. Added OpenPose 1.0.1 portable demo.
-    4. Removed Python and some unnecessary boost dependencies on the VS project.
-    5. Replaced all double quotes by angle brackets in include statements (issue #61).
-    6. Added 3-D reconstruction demo.
-    7. Auto-detection of the camera index.
+    1. Added OpenCV 3.3 compatibility.
+    2. Caffe turned into DLL library.
+    3. OpenPose is now completely portable across Windows 10 computers (with Nvidia graphic card).
+    4. Added OpenPose 1.0.1 portable demo.
+    5. Removed Python and some unnecessary boost dependencies on the VS project.
+    6. Replaced all double quotes by angle brackets in include statements (issue #61).
+    7. Added 3-D reconstruction demo.
+    8. Auto-detection of the camera index.
+    9. Speed up of ~30% in op::floatPtrToUCharCvMat.
+    10. COCO extractor now extracts image ID from the image name itslef (format "string_%d"). Before, only working with validation test, now applicable to e.g. test sets.
+    11. Changed display texts, added `OpenPose` name.
 2. Main bugs fixed:
     1. Pycaffe can now be imported from Python.
     2. Fixed `Tutorial/Wrapper` VS linking errors.

examples/tests/pose_accuracy_coco_test.sh

@@ -19,7 +19,7 @@
 # ./build/examples/openpose/openpose.bin --image_dir "/home/gines/devel/images/val2014" --write_coco_json ../evaluation/coco/results/openpose/1_3.json --no_display --render_pose 0 --scale_number 3 --scale_gap 0.25 --frame_last 3558
 
 #     # 4 scales
-# ./build/examples/openpose/openpose.bin --num_gpu 1 --image_dir "/home/gines/devel/images/val2014" --write_coco_json ../evaluation/coco/results/openpose/1_4.json --no_display --render_pose 0 --scale_number 4 --scale_gap 0.25 --net_resolution "1312x736" --frame_last 3558
+# ./build/examples/openpose/openpose.bin --num_gpu 1 --image_dir "/home/gines/devel/images/val2014" --write_coco_json ../evaluation/coco/results/openpose/1_4.json --no_display --render_pose 0 --num_gpu 1 --scale_number 4 --scale_gap 0.25 --net_resolution "1312x736" --frame_last 3558
 
 # Debugging - Rendered frames saved
 #  ./build/examples/openpose/openpose.bin --image_dir "/home/gines/devel/images/val2014" --write_images ../evaluation/coco/results/openpose/frameOutput --no_display

include/openpose/filestream/cocoJsonSaver.hpp

@@ -20,7 +20,7 @@ namespace op
 
         ~CocoJsonSaver();
 
-        void record(const Array<float>& poseKeypoints, const unsigned long long imageId);
+        void record(const Array<float>& poseKeypoints, const std::string& imageName);
 
     private:
         JsonOfstream mJsonOfstream;

include/openpose/filestream/wCocoJsonSaver.hpp

@@ -60,11 +60,7 @@ namespace op
                 // T* to T
                 const auto& tDatum = tDatums->at(0);
                 // Record json in COCO format
-                const std::string stringToRemove = "COCO_val2014_";
-                const auto stringToRemoveEnd = tDatum.name.find(stringToRemove) + stringToRemove.size();
-                const auto imageId = std::stoull(tDatum.name.substr(stringToRemoveEnd, tDatum.name.size() - stringToRemoveEnd));
-                // Record json in COCO format if file within desired range of images
-                spCocoJsonSaver->record(tDatum.poseKeypoints, imageId);
+                spCocoJsonSaver->record(tDatum.poseKeypoints, tDatum.name);
                 // Profiling speed
                 Profiler::timerEnd(profilerKey);
                 Profiler::printAveragedTimeMsOnIterationX(profilerKey, __LINE__, __FUNCTION__, __FILE__, Profiler::DEFAULT_X);

include/openpose/utilities/string.hpp

@@ -5,6 +5,8 @@
 
 namespace op
 {
+    OP_API unsigned long long getLastNumber(const std::string& string);
+
     /**
      * This template function turns an integer number into a fixed-length std::string.
      * @param number T integer corresponding to the integer to be formatted.

src/openpose/filestream/cocoJsonSaver.cpp

@@ -1,3 +1,4 @@
+#include <openpose/utilities/string.hpp>
 #include <openpose/filestream/cocoJsonSaver.hpp>
 
 namespace op
@@ -27,12 +28,13 @@ namespace op
         }
     }
 
-    void CocoJsonSaver::record(const Array<float>& poseKeypoints, const unsigned long long imageId)
+    void CocoJsonSaver::record(const Array<float>& poseKeypoints, const std::string& imageName)
     {
         try
         {
             const auto numberPeople = poseKeypoints.getSize(0);
             const auto numberBodyParts = poseKeypoints.getSize(1);
+            const auto imageId = getLastNumber(imageName);
             for (auto person = 0 ; person < numberPeople ; person++)
             {
                 // Comma at any moment but first element

src/openpose/utilities/string.cpp

@@ -3,6 +3,20 @@
 
 namespace op
 {
+    unsigned long long getLastNumber(const std::string& string)
+    {
+        try
+        {
+            const auto stringNumber = string.substr(string.find_last_not_of("0123456789") + 1);
+            return std::stoull(stringNumber);
+        }
+        catch (const std::exception& e)
+        {
+            error(e.what(), __LINE__, __FUNCTION__, __FILE__);
+            return 0ull;
+        }
+    }
+
     template<typename T>
     std::string toFixedLengthString(const T number, const unsigned long long stringLength)
     {