提交 6d3fe1e1 编写于 作者: G gineshidalgo99

Fixed CPU bug in 174faa3c

上级 49b8a92a
......@@ -55,18 +55,22 @@ namespace op
if (mCocoJsonFormat == CocoJsonFormat::Body)
{
// Body
if (numberBodyParts == 18)
if (numberBodyParts == 23)
indexesInCocoOrder = std::vector<int>{0, 14,13,16,15, 4,1,5,2,6, 3,10,7,11, 8, 12, 9};
else 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 || numberBodyParts == 25 || numberBodyParts == 59)
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 if (numberBodyParts == 23)
// indexesInCocoOrder = std::vector<int>{18,21,19,22,20, 4,1,5,2,6, 3,13,8,14, 9, 15,10};
}
// Foot
else if (mCocoJsonFormat == CocoJsonFormat::Foot)
{
if (numberBodyParts == 25)
indexesInCocoOrder = std::vector<int>{19,20,21, 22,23,24};
else if (numberBodyParts == 23)
indexesInCocoOrder = std::vector<int>{17,18,19, 20,21,22};
}
// Car
else if (mCocoJsonFormat == CocoJsonFormat::Car)
......
......@@ -65,8 +65,8 @@ namespace op
{
try
{
if (poseModel != PoseModel::BODY_25 || poseModel != PoseModel::COCO_18
|| poseModel != PoseModel::MPI_15 || poseModel != PoseModel::MPI_15_4)
if (poseModel != PoseModel::BODY_25 && poseModel != PoseModel::COCO_18
&& poseModel != PoseModel::MPI_15 && poseModel != PoseModel::MPI_15_4)
error("Model not implemented for CPU body connector.", __LINE__, __FUNCTION__, __FILE__);
// std::vector<std::pair<std::vector<int>, double>> refers to:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册