diff --git a/code/codes/MTCNNv1/generateBoundingBox.m b/code/codes/MTCNNv1/generateBoundingBox.m index 060ff99..dd7e76c 100644 --- a/code/codes/MTCNNv1/generateBoundingBox.m +++ b/code/codes/MTCNNv1/generateBoundingBox.m @@ -2,24 +2,17 @@ %use heatmap to generate bounding boxes stride=2; cellsize=12; - boundingbox=[]; - map=map'; - dx1=reg(:,:,1)'; - dy1=reg(:,:,2)'; - dx2=reg(:,:,3)'; - dy2=reg(:,:,4)'; + dx1=reg(:,:,1); + dy1=reg(:,:,2); + dx2=reg(:,:,3); + dy2=reg(:,:,4); [y x]=find(map>=t); a=find(map>=t); - if size(y,1)==1 - y=y';x=x';score=map(a)';dx1=dx1';dy1=dy1';dx2=dx2';dy2=dy2'; - else - score=map(a); - end + score=map(a); reg=[dx1(a) dy1(a) dx2(a) dy2(a)]; if isempty(reg) reg=reshape([],[0 3]); end - boundingbox=[y x]; + boundingbox=[x y]; boundingbox=[fix((stride*(boundingbox-1)+1)/scale) fix((stride*(boundingbox-1)+cellsize-1+1)/scale) score reg]; end -