提交 708b1b57 编写于 作者: G Gines Hidalgo

Fixed Unity/cudnn bindings/script

Signed-off-by: NGines Hidalgo <gineshidalgo99@gmail.com>
上级 a72618c6
......@@ -3,6 +3,7 @@
echo "This script assumes Ubuntu 16 or 14 and Nvidia Graphics card up to 10XX. Otherwise, it will fail."
# Install cuDNN 5.1
ubuntu_version="$(lsb_release -r)"
if [[ $UBUNTU_VERSION == *"14."* ]] || [[ $UBUNTU_VERSION == *"15."* ]] || [[ $UBUNTU_VERSION == *"16."* ]]; then
CUDNN_URL="http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz"
wget -c ${CUDNN_URL}
......
......@@ -532,6 +532,7 @@ namespace op
OP_API void _OPConfigurePose(
unsigned char poseMode,
int netInputSizeX, int netInputSizeY, // Point
double netInputSizeDynamicBehavior,
int outputSizeX, int outputSizeY, // Point
unsigned char keypointScaleMode, // ScaleMode
int gpuNumber, int gpuNumberStart, int scalesNumber, float scaleGap,
......@@ -547,10 +548,10 @@ namespace op
try
{
spWrapperStructPose = std::make_shared<WrapperStructPose>(
(PoseMode)poseMode, Point<int>{netInputSizeX, netInputSizeY}, Point<int>{outputSizeX, outputSizeY},
(ScaleMode) keypointScaleMode, gpuNumber, gpuNumberStart, scalesNumber, scaleGap,
(RenderMode) renderMode, (PoseModel) poseModel, blendOriginalFrame, alphaKeypoint, alphaHeatMap,
defaultPartToRender, modelFolder,
(PoseMode)poseMode, Point<int>{netInputSizeX, netInputSizeY}, netInputSizeDynamicBehavior,
Point<int>{outputSizeX, outputSizeY}, (ScaleMode) keypointScaleMode, gpuNumber, gpuNumberStart,
scalesNumber, scaleGap, (RenderMode) renderMode, (PoseModel) poseModel, blendOriginalFrame,
alphaKeypoint, alphaHeatMap, defaultPartToRender, modelFolder,
// HeatMapType // (HeatMapType) heatmap_type,
flagsToHeatMaps(heatMapAddParts, heatMapAddBkg, heatMapAddPAFs),
(ScaleMode) heatMapScaleMode, addPartCandidates, renderThreshold, numberPeopleMax,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册