Skip to content

Commit b4a7a6c

Browse files
Changed body_59 colors & allowed COCO saving
1 parent 7e64009 commit b4a7a6c

File tree

2 files changed

+43
-20
lines changed

2 files changed

+43
-20
lines changed

include/openpose/pose/poseParametersRender.hpp

Lines changed: 42 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@ namespace op
117117
7,19, 19,20, 20,21, 21,22, 7,23, 23,24, 24,25, 25,26, 7,27, 27,28, 28,29, 29,30, 7,31, 31,32, 32,33, 33,34, 7,35, 35,36, 36,37, 37,38, \
118118
4,39, 39,40, 40,41, 41,42, 4,43, 43,44, 44,45, 45,46, 4,47, 47,48, 48,49, 49,50, 4,51, 51,52, 52,53, 53,54, 4,55, 55,56, 56,57, 57,58 \
119119
}
120+
// Body + left hand + right hand
120121
#define POSE_BODY_59_COLORS_RENDER_GPU \
121122
255.f, 0.f, 85.f, \
122123
255.f, 0.f, 0.f, \
@@ -138,25 +139,47 @@ namespace op
138139
255.f, 0.f, 255.f, \
139140
85.f, 0.f, 255.f, \
140141
\
141-
255.f, 0.f, 85.f, \
142-
255.f, 0.f, 0.f, \
143-
255.f, 85.f, 0.f, \
144-
255.f, 170.f, 0.f, \
145-
255.f, 255.f, 0.f, \
146-
170.f, 255.f, 0.f, \
147-
85.f, 255.f, 0.f, \
148-
0.f, 255.f, 0.f, \
149-
255.f, 0.f, 0.f, \
150-
0.f, 255.f, 85.f, \
151-
0.f, 255.f, 170.f, \
152-
0.f, 255.f, 255.f, \
153-
0.f, 170.f, 255.f, \
154-
0.f, 85.f, 255.f, \
155-
0.f, 0.f, 255.f, \
156-
255.f, 0.f, 170.f, \
157-
170.f, 0.f, 255.f, \
158-
255.f, 0.f, 255.f, \
159-
85.f, 0.f, 255.f
142+
100.f, 0.f, 0.f, \
143+
150.f, 0.f, 0.f, \
144+
200.f, 0.f, 0.f, \
145+
255.f, 0.f, 0.f, \
146+
100.f, 100.f, 0.f, \
147+
150.f, 150.f, 0.f, \
148+
200.f, 200.f, 0.f, \
149+
255.f, 255.f, 0.f, \
150+
0.f, 100.f, 50.f, \
151+
0.f, 150.f, 75.f, \
152+
0.f, 200.f, 100.f, \
153+
0.f, 255.f, 125.f, \
154+
0.f, 50.f, 100.f, \
155+
0.f, 75.f, 150.f, \
156+
0.f, 100.f, 200.f, \
157+
0.f, 125.f, 255.f, \
158+
100.f, 0.f, 100.f, \
159+
150.f, 0.f, 150.f, \
160+
200.f, 0.f, 200.f, \
161+
255.f, 0.f, 255.f, \
162+
\
163+
100.f, 0.f, 0.f, \
164+
150.f, 0.f, 0.f, \
165+
200.f, 0.f, 0.f, \
166+
255.f, 0.f, 0.f, \
167+
100.f, 100.f, 0.f, \
168+
150.f, 150.f, 0.f, \
169+
200.f, 200.f, 0.f, \
170+
255.f, 255.f, 0.f, \
171+
0.f, 100.f, 50.f, \
172+
0.f, 150.f, 75.f, \
173+
0.f, 200.f, 100.f, \
174+
0.f, 255.f, 125.f, \
175+
0.f, 50.f, 100.f, \
176+
0.f, 75.f, 150.f, \
177+
0.f, 100.f, 200.f, \
178+
0.f, 125.f, 255.f, \
179+
100.f, 0.f, 100.f, \
180+
150.f, 0.f, 150.f, \
181+
200.f, 0.f, 200.f, \
182+
255.f, 0.f, 255.f
160183

161184
// Rendering functions
162185
OP_API const std::vector<float>& getPoseColors(const PoseModel poseModel);

src/openpose/filestream/cocoJsonSaver.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ namespace op
6969
std::vector<int> indexesInCocoOrder;
7070
if (numberBodyParts == 18)
7171
indexesInCocoOrder = std::vector<int>{0, 15,14,17,16, 5,2,6,3,7, 4,11,8,12, 9, 13,10};
72-
else if (numberBodyParts == 19)
72+
else if (numberBodyParts == 19 || numberBodyParts == 59)
7373
indexesInCocoOrder = std::vector<int>{0, 16,15,18,17, 5,2,6,3,7, 4,12,9,13,10, 14,11};
7474
else if (numberBodyParts == 23)
7575
indexesInCocoOrder = std::vector<int>{18,21,19,22,20, 4,1,5,2,6, 3,13,8,14, 9, 15,10};

0 commit comments

Comments
 (0)