zerollzeng / tensorrt-zoo

openpose, yolov3 with tiny-tensorrt
86 stars 25 forks source link

changing this project to coco_ 18 key points predicted #37

Closed weilanShi closed 4 years ago

weilanShi commented 4 years ago

Who changed this project to coco_18 keypoints predicted and work successful? i changed the mNumPeaks (openpose.hpp) to 18, and the net_output.resize(78 60 80) to net_output.resize(57 60 80) (openpose.cu) and the op::PoseModel::BODY_25 to op::PoseModel::COCO_18 (op::connectBodyPartsCpu of openpose.cu ) and changed the const auto& bodyPartPairs and const auto numberBodyParts (void connectBodyPartsCpu of BodyPartConnector.cu). Unfortunately, there is none in the result.jpg. Anyone could help me, thanks!

zerollzeng commented 4 years ago

All the post-processing was ported from https://github.com/CMU-Perceptual-Computing-Lab/openpose/tree/master/src/openpose/net which might be helpful to you

zerollzeng commented 4 years ago

Closed due to inativity