part based model 模型 mat结构模型转换为opencv中的xml的格式

链接:http://www.csie.ntu.edu.tw/~r99922070/code/

代码:matlab格式

% gtyu 2012-06-12
% National Taiwan University
% r99922070@ntu.edu.tw

% Convert *.mat format model in the source example in 
% Discriminatively Trained Deformable Part Models "voc-release4.01"
% to opencv's latentSVM detect input format *.xml

function MAT2XMLmodel_401(matmodel, xmlfile)

matmodel = 'VOC2009/cat_final';
xmlfile = 'VOC2009/cat_final.xml';

load(matmodel);
fid = fopen(xmlfile, 'w');

fprintf(fid, '<Model>\n');

ncom = length(model.rules{model.start});
fprintf(fid, '\t<!-- Number of components -->\n');
fprintf(fid, '\t<NumComponents>%d</NumComponents>\n', ncom);

nfeature = 31;
fprintf(fid, '\t<!-- Number of features -->\n');
fprintf(fid, '\t<P>%d</P>\n', nfeature);

fprintf(fid, '\t<!-- Score threshold -->\n');
fprintf(fid, '\t<ScoreThreshold>%.16f</ScoreThreshold>\n', model.thresh);
layer = 1;
for icom = 1:ncom
    fprintf(fid, '\t<Component>\n');
    
        fprintf(fid, '\t\t<!-- Root filter description -->\n');
        fprintf(fid, '\t\t<RootFilter>\n');
        
            % attention: X,Y swap
            rhs = model.rules{model.start}(icom).rhs;
            % assume the root filter is first on the rhs of the start rules
            if model.symbols(rhs(1)).type == 'T'
              % handle case where there's no deformation model for the root
              root = model.symbols(rhs(1)).filter;
            else
              % handle case where there is a deformation model for the root
              root = model.symbols(model.rules{rhs(1)}(layer).rhs).filter;
            end
            
            filternum = root;
            sizeX = model.filters(filternum).size(2);
            sizeY = model.filters(filternum).size(1);
            fprintf(fid, '\t\t\t<!-- Dimensions -->\n'); 
            fprintf(fid, '\t\t\t<sizeX>%d</sizeX>\n', sizeX); 
            fprintf(fid, '\t\t\t<sizeY>%d</sizeY>\n', sizeY);
        
            fprintf(fid, '\t\t\t<!-- Weights (binary representation) -->\n');
            fprintf(fid, '\t\t\t<Weights>');
            for iY = 1:sizeY
                for iX = 1:sizeX
                    % original mat has 32 which is larger than nfeature=31 by 1
                    fwrite(fid, model.filters(filternum).w(iY,iX,1:nfeature), 'double'); % need verify
                end
            end
            fprintf(fid, '\t\t\t</Weights>\n');
            
    		
            fprintf(fid, '\t\t\t<!-- Linear term in score function -->\n');
            fprintf(fid, '\t\t\t<LinearTerm>%.16f</LinearTerm>\n', ...  % need verify
                model.rules{model.start}(icom).offset.w);
            
        fprintf(fid, '\t\t</RootFilter>\n');
        
        fprintf(fid, '\t\t<!-- Part filters description -->\n');
        fprintf(fid, '\t\t<PartFilters>\n');
        
        npart = length(model.rules{model.start}(icom).rhs) -1 ;
        fprintf(fid, '\t\t\t<NumPartFilters>%d</NumPartFilters>\n', npart);
        
        for ipart = 2: npart+1
            fprintf(fid, '\t\t\t<!-- Part filter ? description -->\n');
            fprintf(fid, '\t\t\t<PartFilter>\n');
            
            irule = model.rules{model.start}(icom).rhs(ipart);
            filternum = model.symbols(model.rules{irule}.rhs).filter;
            sizeX = model.filters(filternum).size(2);
            sizeY = model.filters(filternum).size(1);
            fprintf(fid, '\t\t\t\t<sizeX>%d</sizeX>\n', sizeX);
            fprintf(fid, '\t\t\t\t<sizeY>%d</sizeY>\n', sizeY);
            fprintf(fid, '\t\t\t\t<!-- Weights (binary representation) -->\n');
            fprintf(fid, '\t\t\t\t<Weights>');
            for iY = 1:sizeY
                for iX = 1:sizeX
                    % original mat has 32 which is larger than nfeature=31 by 1
                    fwrite(fid, model.filters(filternum).w(iY,iX,1:nfeature), 'double'); % need verify
                end
            end
            fprintf(fid, '\t\t\t\t</Weights>\n');
            
            fprintf(fid, '\t\t\t\t<!-- Part filter offset -->\n');
            
            fprintf(fid, '\t\t\t\t<V>\n');
            
            fprintf(fid, '\t\t\t\t\t<Vx>%d</Vx>\n',model.rules{model.start}(icom).anchor{ipart}(1)+1); %[dx,dy,ds]
            fprintf(fid, '\t\t\t\t\t<Vy>%d</Vy>\n',model.rules{model.start}(icom).anchor{ipart}(2)+1);
            
            fprintf(fid, '\t\t\t\t</V>\n');
            
            
            fprintf(fid, '\t\t\t\t<!-- Quadratic penalty function coefficients -->\n');
            
            fprintf(fid, '\t\t\t\t<Penalty>\n');
            fprintf(fid, '\t\t\t\t\t<dx>%.16f</dx>\n',model.rules{irule}.def.w(2)); 
            fprintf(fid, '\t\t\t\t\t<dy>%.16f</dy>\n',model.rules{irule}.def.w(4)); 
            fprintf(fid, '\t\t\t\t\t<dxx>%.16f</dxx>\n',model.rules{irule}.def.w(1)); 
            fprintf(fid, '\t\t\t\t\t<dyy>%.16f</dyy>\n',model.rules{irule}.def.w(3)); 
            
            fprintf(fid, '\t\t\t\t</Penalty>\n');
				
            
            
            fprintf(fid, '\t\t\t</PartFilter>\n');
        end
    
        fprintf(fid, '\t\t</PartFilters>\n');
    fprintf(fid, '\t</Component>\n');
end
fprintf(fid, '</Model>\n');
fclose(fid);

end

注意:代码在linux环境下运行成功,windows下运行该代码貌似有问题。
转换后的模型可以再opencv当中使用。
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值