MyJob* JobServer::getJob(std::string s)
{
std::stringstream ss;
std::string line;
ss << s;
getline(ss, line, ' '); //skip the issuejob
getline(ss, line, ' '); //get the actual name
std::string name = line; //set it in a string
int argc = getLastNumber(s); //get the number of arguments
char** c = new char*[argc+1]; //create an array for them
c[argc] = NULL; //set the last entry of it as null
c[0] = new char(name.length() + 1); //and the first one as the name given
strcpy(c[0], line.c_str());
for(int i = 1; i < argc; i++) //for each other argument
{
getline(ss, line, ' '); //get it
c[i] = new char[line.length() + 1]; //and put it in the array
strcpy(c[i], line.c_str());
} //create the new job (pid, job number, name, arguments, number of arguments)
MyJob* mj = new MyJob(0, jobCount, name, c, argc);
return mj; //return it
}
void CocoJsonSaver::record(const Array<float>& poseKeypoints, const Array<float>& poseScores,
const std::string& imageName)
{
try
{
// Security checks
if ((size_t)poseKeypoints.getSize(0) != poseScores.getVolume())
error("Dimension mismatch between poseKeypoints and poseScores.", __LINE__, __FUNCTION__, __FILE__);
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
if (mFirstElementAdded)
{
mJsonOfstream.comma();
mJsonOfstream.enter();
}
else
mFirstElementAdded = true;
// New element
mJsonOfstream.objectOpen();
// image_id
mJsonOfstream.key("image_id");
mJsonOfstream.plainText(imageId);
mJsonOfstream.comma();
// category_id
mJsonOfstream.key("category_id");
mJsonOfstream.plainText("1");
mJsonOfstream.comma();
// keypoints - i.e. poseKeypoints
mJsonOfstream.key("keypoints");
mJsonOfstream.arrayOpen();
std::vector<int> indexesInCocoOrder;
if (numberBodyParts == 18)
indexesInCocoOrder = std::vector<int>{0, 15,14,17,16, 5,2,6,3,7, 4,11,8,12, 9, 13,10};
else if (numberBodyParts == 19)
indexesInCocoOrder = std::vector<int>{0, 16,15,18,17, 5,2,6,3,7, 4,12,9,13,10, 14,11};
else if (numberBodyParts == 23)
indexesInCocoOrder = std::vector<int>{18,21,19,22,20, 4,1,5,2,6, 3,13,8,14, 9, 15,10};
else
error("Unvalid number of body parts (" + std::to_string(numberBodyParts) + ").",
__LINE__, __FUNCTION__, __FILE__);
for (auto bodyPart = 0u ; bodyPart < indexesInCocoOrder.size() ; bodyPart++)
{
const auto finalIndex = 3*(person*numberBodyParts + indexesInCocoOrder.at(bodyPart));
mJsonOfstream.plainText(poseKeypoints[finalIndex]);
mJsonOfstream.comma();
mJsonOfstream.plainText(poseKeypoints[finalIndex+1]);
mJsonOfstream.comma();
mJsonOfstream.plainText((poseKeypoints[finalIndex+2] > 0.f ? 1 : 0));
// mJsonOfstream.plainText(poseKeypoints[finalIndex+2]); // For debugging
if (bodyPart < indexesInCocoOrder.size() - 1u)
mJsonOfstream.comma();
}
mJsonOfstream.arrayClose();
mJsonOfstream.comma();
// score
mJsonOfstream.key("score");
mJsonOfstream.plainText(poseScores[person]);
mJsonOfstream.objectClose();
}
}
catch (const std::exception& e)
{
error(e.what(), __LINE__, __FUNCTION__, __FILE__);
}
}
