poselets-lite/hit_list.m at master · bensapp/poselets-lite · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
classdef hit_list
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%%%
%%% Represents a list of hits (poselet/torso/object_bounds hits) in one or
%%% more images
%%%
%%% Copyright (C) 2009, Lubomir Bourdev and Jitendra Malik.
%%% This code is distributed with a non-commercial research license.
%%% Please see the license file license.txt included in the source directory.
%%%
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%

    properties
        bounds;      % Nx4 array of single. [min_x min_y width height]
        poselet_id;  % Nx1 arrat of uint16. Unique ID of the detected poselet
        score;       % Nx1 array of single. Score of the SVM classifier
        image_id;    % Nx1 array of uint16. Unique ID of the image that contains the hit
        src_idx;     % Nx1 cell of uint32 sets. For each hit, the indices of the source hits. (for example, for a torso, indices of the poselets that voted for it)
        size;        % uint32 size
    end
    methods
        function h = hit_list(bounds,score,poselet_id,image_id,src_idx)
            if nargin==0
                h.size=0;
                h.bounds=zeros(4,0,'single');
                h.poselet_id=zeros(0,1,'uint16');
                h.score=zeros(0,1,'single');
                h.image_id=zeros(0,1,'uint16');
                h.src_idx = cell(0,0);
            else
                assert(size(bounds,1)==4);
                h.size=size(bounds,2);
                h.bounds=single(bounds);
                h.score(1:h.size,1)=single(score);
                h.poselet_id(1:h.size,1)=uint16(poselet_id);
                h.image_id(1:h.size,1)=uint16(image_id);
                if exist('src_idx','var')
                   h.src_idx = src_idx;
                else
                   h.src_idx = cell(0,0);
                end
            end

        end

        function is=isempty(h)
           is=(h.size==0);
        end

        %%% Appends hits h2 at the end of hits h
        function h=append(h,h2)
            if isempty(h2)
                return;
            end
            rng=1:h2.size;
            h.bounds(:,h.size+rng)      = h2.bounds(:,rng);
            h.score(h.size+rng,:)       = h2.score(rng);
            h.image_id(h.size+rng,:)    = h2.image_id(rng);
            h.poselet_id(h.size+rng,:)  = h2.poselet_id(rng);
            if ~isempty(h2.src_idx)
                h.src_idx(h.size+rng,:)     = h2.src_idx(rng);
            else
                h.src_idx = {};
            end
            h.size=h.size+h2.size;
        end

        function h=reserve(h,len)
            diff = len-length(h.score);
            if diff>0
               h.bounds(1:4,end+(1:diff)) = nan;
               h.score(end+(1:diff),1)=nan;
               h.image_id(end+(1:diff),1)=0;
               h.poselet_id(end+(1:diff),1)=0;
               h.src_idx{len,1}=[];
            end
        end

        %%% Returns a subset of the input hits
        function h=select(h,sel)
            if isempty(sel)
                h=hit_list;
                return;
            end
            h.bounds = h.bounds(:,sel);
            h.score = h.score(sel,1);
            h.image_id = h.image_id(sel,1);
            h.poselet_id = h.poselet_id(sel,1);
            if ~isempty(h.src_idx)
                h.src_idx = h.src_idx(sel,1);
            end
            h.size=length(h.score);
        end

        %%% Draws the rotated bounding box of a hit
        function draw_bounds(h,color,linewidth,linestyle,textbg,textfg)
            if ~exist('color','var')
                color='g';
            end
            if ~exist('linewidth','var')
               linewidth=0.5;
            end
            if ~exist('linestyle','var')
               linestyle='-';
            end

            for i=1:h.size
                rectangle('position',h.bounds(:,i),'edgecolor',color,'linewidth',linewidth,'linestyle',linestyle);
                if exist('textbg','var')
                    if ~exist('textfg','var')
                       textfg = [1 1 1];
                    end
                    text(double(h.bounds(1,i)),double(h.bounds(2,i)),num2str(h.score(i),'%4.2f'),'BackgroundColor',textbg,'Color',textfg);
                end
            end
        end
    end
end