Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields ∗ 源代码

Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields ∗ 源代码　open pose 实时人体姿态估计 caffe+python+matlab

function param = config（id）
%% set this part

% GPU device number （doesn‘t matter for CPU mode）
GPUdeviceNumber = 0;

% Select model （default: 5）
param.modelID = id;

% Use click mode or not. If yes （1） you will be asked to click on the center
% of person to be pose-estimated （for multiple people image）. If not （0）
% the model will simply be applies on the whole image.
param.click = 1;

% Scaling paramter: starting and ending ratio of person height to image
% height and number of scales per octave
% warning: setting too small starting value on non-click mode will take
% large memory

% CPU mode or GPU mode
param.use_gpu = 1;

param.test_mode = 3;
param.vis = 1;

param.octave = 6;
param.starting_range = 0.8;
param.ending_range = 2;

param.min_num = 4;
param.mid_num = 10;

% the larger the crop_ratio the smaller the windowsize
param.crop_ratio = 2.5; %2
param.bbox_ratio = 0.25; %0.5

% applyModel_max
param.max = 0;
% use average heatmap
param.merge = ‘avg‘;


% path of your caffe
caffepath = ‘/home/zhecao/caffe/matlab‘;

%COCO parameter
if id == 1
    param.scale_search = [0.5 1 1.5 2];
    param.thre1 = 0.1;
    param.thre2 = 0.05; 
    param.thre3 = 0.5; 

    param.model（id）.caffemodel = ‘../model/_trained_COCO/pose_iter_440000.caffemodel‘;
    param.model（id）.deployFile = ‘../model/_trained_COCO/pose_deploy.prototxt‘;
    param.model（id）.description = ‘COCO Pose56 Two-level Linevec‘;
    param.model（id）.boxsize = 368;
    param.model（id）.padValue = 128;
    param.model（id）.np = 18; 
    param.model（id）.part_str = {‘nose‘ ‘neck‘ ‘Rsho‘ ‘Relb‘ ‘Rwri‘ ... 
                             ‘Lsho‘ ‘Lelb‘ ‘Lwri‘ ...
                             ‘Rhip‘ ‘Rkne‘ ‘Rank‘ ...
                             ‘Lhip‘ ‘Lkne‘ ‘Lank‘ ...
                             ‘Leye‘ ‘Reye‘ ‘Lear‘ ‘Rear‘ ‘pt19‘};
end

%MPI parameter
if id == 2
    param.scale_search = [0.7 1 1.3];
    param.thre1 = 0.05;
    param.thre2 = 0.01;
    param.thre3 = 3;
    param.thre4 = 0.1;
    
    param.model（id）.caffemodel = ‘../model/_trained_MPI/pose_iter_146000.caffemodel‘;
    param.model（id）.deployFile = ‘../model/_trained_MPI/pose_deploy.prototxt‘;
    
    param.model（id）.description = ‘MPI Pose43 Two-level LineVec‘;
    param.model（id）.boxsize = 368;
    param.model（id）.padValue = 128;
    param.model（id）.np = 15; 
    param.model（id）.part_str = {‘Nose‘ ‘Neck‘ ‘Rsho‘ ‘Relb‘ ‘Rwri‘ ... 
                             ‘Lsho‘ ‘Lelb‘ ‘Lwri‘ ...
                             ‘Rhip‘ ‘Rkne‘ ‘Rank‘ ...
                             ‘Lhip‘ ‘Lkne‘ ‘Lank‘ ‘center‘};
end


disp（caffepath）;
addpath（caffepath）;
caffe.set_mode_gpu（）;
caffe.set_device（GPUdeviceNumber）;
caffe.reset_all（）;

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\
     文件          18  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\.gitignore
     文件           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\.gitmodules
     文件        9686  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\LICENSE
     文件        5258  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\README.md
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\model\
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\model\_trained_COCO\
     文件       46106  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\model\_trained_COCO\pose_deploy.prototxt
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\model\_trained_MPI\
     文件       46108  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\model\_trained_MPI\pose_deploy.prototxt
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\readme\
     文件      974405  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\readme\arch.png
     文件     2086052  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\readme\dance.gif
     文件     2887504  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\readme\pose.gif
     文件     2040076  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\readme\shake.gif
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\
     文件        2708  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\config.m
     文件         708  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\demo.m
     文件        2814  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\evalCOCO.m
     文件        3113  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\evalMPII.m
     文件         262  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\get_model.sh
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\.ipynb_checkpoints\
     文件    10975518  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\.ipynb_checkpoints\demo-checkpoint.ipynb
     文件         772  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\config
     文件        1189  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\config_reader.py
     文件     5927062  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\demo.ipynb
     文件        2642  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\python\util.py
     目录           0  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\sample_image\
     文件      305923  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\sample_image\ski.jpg
     文件      646367  2017-11-29 21:42  Realtime_Multi-Person_Pose_Estimation-master\testing\sample_image\upper.jpg
............此处省略98个文件信息