lhireplaceobjectname.m

来自「This code can parse any image in matlab.」· M 代码 · 共 60 行

function D = LHIreplaceobjectname(D, name1, name2, method, querymethod)
% Replace an object name by another name.
% It is not case sensitive.
%
% D = LHIreplaceobjectname(D, 'person walking', 'pedestrian')
%
% Reemplaces all the object names that contain the string 'person' by the
% short name 'person'
% D = LHIreplaceobjectname(D, 'person', 'person', 'rename')
%
% To prevent mistake replacement (e.g. intepretate 'basket'-> 'basketball'), 
% use querymethod  'word'
% D = LHIreplaceobjectname(D, 'bakset', 'container', 'rename','word') 
% Detail for querymethod see LHIquery.m
%
% Reemplaces all the object names that contain the strings 'person' or 'pedestrian' by the
% short name 'person'
% D = LHIreplaceobjectname(D, 'person,pedestrian', 'person', 'rename')
%
% It does not modify the annotation files. It only modifies the index
% struct.

if nargin < 4
    method = 'replace';
end

name1 = lower(name1);
name2 = lower(name2);

if nargin < 5
    [d, j] = LHIquery(D, 'object.name', name1);
else
    [d, j] = LHIquery(D, 'object.name', name1, querymethod);
end
clear d

Nimages = length(j);
for n = 1:Nimages
    %Nobjects = length(D(j(n)).annotation.object);
    if nargin < 5
        jc = LHIobjectindex(D(j(n)).annotation, name1);
    else
        jc = LHIobjectindex(D(j(n)).annotation, name1, querymethod);
    end
    
    for m = jc
        currentname = lower(D(j(n)).annotation.object(m).name);
        switch method
            case 'replace'
                % replace each substring name1 by name2
                D(j(n)).annotation.object(m).name = strrep(currentname, name1, name2);
            case 'rename'
                % rename the object
                D(j(n)).annotation.object(m).name = name2;
        end
    end
end

disp(sprintf('%d entries replaced', Nimages))