参见人群密度估计之MCNN写的关于密度图的生成,在MATLAB跑了程序,代码如下:
clear;
%透视变换的系数
load('perspective_roi.mat');
%ground truth
load('mall_gt.mat');
m=480;n=640;
m=m/4;
n=n/4;
mask = imresize(roi.mask,0.25); %图像缩小4倍
for i=1:2000 %2000幅图像
gt = frame{i}.loc; %第一个frame结构体的loc字段
gt = gt/4;
d_map = zeros(m,n);
for j=1:size(gt,1)
%设置高斯滤波的参数
ksize = ceil(25/sqrt(pMapN(floor(gt(j,2)),1)));
ksize = max(ksize,7);
ksize = min(ksize,25);
radius = ceil(ksize/2);
sigma = ksize/2.5