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

Fixed Unity/cudnn bindings/script

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