欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

matlab 图片RGB颜色提取(第二版)

程序员文章站 2022-05-20 21:21:46
...

之前使用matlab写了一个程序能实现导入图片识别颜色
为其增添了一些新的功能,包括:RGB格式设置,各类储存
并扩展了展示框颜色的存放量
新界面(改动不大):matlab 图片RGB颜色提取(第二版)
先介绍一下基础使用方法:
从电脑导入图片或使用截图功能导入图片:
matlab 图片RGB颜色提取(第二版)
matlab 图片RGB颜色提取(第二版)
点击获取颜色后,鼠标在图片上移动时,颜色信息会在这里显示:matlab 图片RGB颜色提取(第二版)
点击储存颜色后颜色会暂时储存在右侧:
matlab 图片RGB颜色提取(第二版)
新版本增添了改变RGB格式的功能:
matlab 图片RGB颜色提取(第二版)
matlab 图片RGB颜色提取(第二版)
matlab 图片RGB颜色提取(第二版)

展示框容量随着取色数增大:

matlab 图片RGB颜色提取(第二版)
新增了四种导出格式:
matlab 图片RGB颜色提取(第二版)
1导出为mat
matlab 图片RGB颜色提取(第二版)
2导出为txt(有待改进)
matlab 图片RGB颜色提取(第二版)
3导出为excel
matlab 图片RGB颜色提取(第二版)
4-1导出当前页面对照图:
matlab 图片RGB颜色提取(第二版)
4-2导出全部页面对照图:
matlab 图片RGB颜色提取(第二版)
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
完整代码

function getcolor
global control;
global ima;
global GUI;
global x_limit;
global y_limit;
global color;
global a;
global color_number;
global color_list;
global page;
global total_page;
global color_list_temp;
color=[];
page=1;
total_page=2;
color_list=[0 0 0];
color_list(1,:)=[];
color_number=1;
rgb_type=1;
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
GUI.fig=figure('units','pixels',...
    'position',[350 100 800 500],...
    'Numbertitle','off',...
    'menubar','none',...
    'resize','off',...
    'name','getcolor',...
    'color',[0.95 0.95 0.95]);
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
uh1=uimenu('label','设置');
uimenu(uh1,'label','RGB类型设置','callback',@RGBset)
    function RGBset(~,~)
        GUI.rgbfig=figure('units','pixels',...
        'position',[360 370 180 200],...
        'Numbertitle','off',...
        'menubar','none',...
        'name','RGBset',...
        'resize','off');
        GUI.axes=axes('Units','pixels',...
         'parent',GUI.rgbfig,...  
        'PlotBoxAspectRatio',[1 1 1],...
        'Color',[0.95 0.95 0.95],...
        'Box','on', ...
        'XLim',[0 500],...
        'YLim',[0 500], ...
        'XColor',[0.95 0.95 0.95],...
        'YColor',[0.95 0.95 0.95],...
        'YDir','reverse', ...
        'xtick',[],'ytick',[]);
        GUI.checkbox1=uicontrol('parent',GUI.rgbfig,...
            'style','checkbox',...
            'string','范围:0-1',...
            'position',[45 150 400 30],...
            'fontsize',10,...
            'value',rgb_type,...
            'callback',@ifon1);
        GUI.checkbox255=uicontrol('parent',GUI.rgbfig,...
            'style','checkbox',...
            'string','范围:0-255',...
            'position',[45 120 400 30],...
            'fontsize',10,...
            'value',~rgb_type,...
            'callback',@ifon255);
        GUI.makesurebutton=uicontrol('parent',GUI.rgbfig,...
            'style','pushbutton',...
            'string','确定设置',...
            'position',[45 70 100 25],...
            'fontsize',10,...
            'callback',@settype);
        
        function ifon1(~,~)
            if(get(GUI.checkbox1,'value')==1)
                set(GUI.checkbox1,'value',1);
                set(GUI.checkbox255,'value',0);
            else
                set(GUI.checkbox1,'value',1);
            end               
        end
        function ifon255(~,~)
            if(get(GUI.checkbox255,'value')==1)
                set(GUI.checkbox255,'value',1);
                set(GUI.checkbox1,'value',0);
            else
                set(GUI.checkbox255,'value',1);
            end                
        end
        function settype(~,~)
            rgb_type=get(GUI.checkbox1,'value');
            if ~isempty(color)
                set(GUI.text2,'string',['[',num2str((color/255).*rgb_type+color.*(~rgb_type)),']']);
            end
            show_color(page);
            close(GUI.rgbfig)
        end      
    end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
uh2=uimenu('label','保存');
uimenu(uh2,'label','储存为mat','callback',@saveas_mat)
uimenu(uh2,'label','储存为txt','callback',@saveas_txt)
uimenu(uh2,'label','储存为excel','callback',@saveas_exl)
uimenupic=uimenu(uh2,'label','储存为对照图');
uimenu(uimenupic,'label','储存全部页码','callback',@saveas_pic_all);
uimenu(uimenupic,'label','储存当前页码','callback',@saveas_pic_now);
function saveas_mat(~,~)
    try
    [filename, pathname] = uiputfile({'*.mat','mat'});
        color_list_temp=(color_list/255).*rgb_type+color_list.*(~rgb_type);
        save([pathname,filename],'color_list_temp');
    catch
    end  
end
function saveas_txt(~,~)
    try 
    [filename, pathname] = uiputfile({'*.txt','记事本'});
        color_list_temp=(color_list/255).*rgb_type+color_list.*(~rgb_type);
        [m,n]=size(color_list_temp);
        fid=fopen([ pathname,filename],'w');
        for ii=1:m
            for jj=1:n
                if jj==n
                    fprintf(fid,'%d\r\n',color_list_temp(ii,jj));
                else
                    fprintf(fid,'%d\r\t',color_list_temp(ii,jj));
                end
            end
        end
        fclose(fid);
    catch
    end
end
function saveas_exl(~,~)
    [filename, pathname] = uiputfile({'*.xlsx','记事本'});
    color_list_temp=(color_list/255).*rgb_type+color_list.*(~rgb_type);
    xlswrite([ pathname,filename],color_list_temp) 
    
end
function saveas_pic_all(~,~)
    page_with_color=total_page-1;
    px=50;
    gap_px=10;
    pic=ones(9*px,page_with_color*px+(page_with_color-1)*gap_px,3);
    for p=1:page_with_color
        for ii=(p-1)*9+1:p*9
            for kk=1:3
                if ii<=length(color_list)
                    pic((ii-(p-1)*9-1)*px+1:(ii-(p-1)*9)*px,(p-1)*(px+gap_px)+1:(p-1)*(px+gap_px)+px,kk)=color_list(ii,kk)/255;     
                end
            end
        end
    end
    [filename, pathname] = uiputfile({'*.jpg;*.png','All Image Files';...
            '*.jpg','JPG';'*.png','PNG' });
    imwrite(pic,[pathname,filename]);
end
function saveas_pic_now(~,~)
    try
    [m,~]=size(color_list);
    m=m-(page-1)*9;
    m(m>9)=9;
    px=50;
    pic=ones(9*px,1*px,3);
    if m>0
        for ii=(page-1)*9+1:(page-1)*9+m
            for kk=1:3
                pic((ii-1)*px+1:ii*px,1:px,kk)=color_list(ii,kk)/255;      
            end
        end
    else
    end
    [filename, pathname] = uiputfile({'*.jpg;*.png','All Image Files';...
            '*.jpg','JPG';'*.png','PNG' });
    imwrite(pic,[pathname,filename]);
    catch
    end
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%uh3=uimenu('label','导入');
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
GUI.text=uicontrol('parent',GUI.fig,...
    'style','text',...
    'string','色彩识别',...
    'horizontalalign','center',...
    'position',[50 440 400 30],...
    'backgroundcolor',[0.85 0.89 0.85],...
    'foregroundcolor','k',...
    'fontsize',15);
GUI.text1=uicontrol('parent',GUI.fig,...
    'style','text',...
    'string','',...
    'horizontalalign','center',...
    'position',[460 330 100 100],...
    'backgroundcolor',[1 1 1],...
    'foregroundcolor','k',...
    'fontsize',10);
GUI.text2=uicontrol('parent',GUI.fig,...
    'style','text',...
    'string','',...
    'horizontalalign','center',...
    'position',[350 440 210 30],...
    'backgroundcolor',[1 1 1],...
    'foregroundcolor','k',...
    'fontsize',10);
GUI.savecolorbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','储存颜色',...
    'position',[460 290 100 30],...
    'backgroundcolor',[0.85 0.89 0.85],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@save_color);
GUI.deletedatabutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','清空数据',...
    'position',[460 230 100 30],...
    'backgroundcolor',[0.8 0.9 0.9],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@clear_data);
GUI.deletepicbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','删除图片',...
    'position',[460 180 100 30],...
    'backgroundcolor',[0.8 0.9 0.9],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@delete_pic);
GUI.getcapbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','屏幕截图',...
    'position',[460 130 100 30],...
    'backgroundcolor',[0.8 0.9 0.9],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@get_capture);
GUI.getpicbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','读取图片',...
    'position',[460 80 100 30],...
    'backgroundcolor',[0.8 0.9 0.9],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@getImage);
GUI.getcolorbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'tag','recc',...
    'string','获取颜色',...
    'position',[460 30 100 30],...
    'backgroundcolor',[0.8 0.9 0.9],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@get_color);
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
for i=1:9
    GUI.text=uicontrol('parent',GUI.fig,...
    'tag',num2str(i),...
    'style','text',...
    'string','',...
    'horizontalalign','left',...
    'position',[600 440-40*(i-1) 30 30],...
    'backgroundcolor',[1 1 1],...
    'foregroundcolor','k',...
    'fontsize',10);    
end

for i=1:9
    GUI.text=uicontrol('parent',GUI.fig,...
    'tag',[num2str(i),'t'],...
    'style','text',...
    'string','',...
    'horizontalalign','center',...
    'position',[640 440-40*(i-1) 150 30],...
    'backgroundcolor',[1 1 1],...
    'foregroundcolor','k',...
    'fontsize',8);    
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
GUI.inputbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','清除最后一个颜色',...
    'position',[600 80 190 30],...
    'backgroundcolor',[0.85 0.89 0.85],...
    'foregroundcolor','k',...
    'fontsize',15,...
    'callback',@delete_last);
%GUI.inputbutton=uicontrol('parent',GUI.fig,...
    %'style','pushbutton',...
    %'string','导出数据',...
    %'position',[600 30 190 30],...
    %'backgroundcolor',[0.85 0.89 0.85],...
    %'foregroundcolor','k',...
    %'fontsize',15,...
    %'callback',@output_data);
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
GUI.lpbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','<上一页',...
    'position',[600 30 70 30],...
    'backgroundcolor',[0.85 0.85 0.85],...
    'foregroundcolor','k',...
    'fontsize',12,...
    'callback',@lastpage);
GUI.npbutton=uicontrol('parent',GUI.fig,...
    'style','pushbutton',...
    'string','下一页>',...
    'position',[720 30 70 30],...
    'backgroundcolor',[0.85 0.85 0.85],...
    'foregroundcolor','k',...
    'fontsize',12,...
    'callback',@nextpage);
GUI.page=uicontrol('parent',GUI.fig,...
    'style','text',...
    'string',[num2str(page),'/',num2str(total_page)],...
    'horizontalalign','center',...
    'position',[670 30 50 27],...
    'backgroundcolor',[0.95 0.95 0.95],...
    'foregroundcolor','k',...
    'fontsize',12);
function lastpage(~,~)
    page=page-1;
    page(page<1)=1;
    set(GUI.page,'string',[num2str(page),'/',num2str(total_page)]);
    show_color(page);
end
function nextpage(~,~)
    page=page+1;
    page(page>total_page)=total_page;
    set(GUI.page,'string',[num2str(page),'/',num2str(total_page)]);
    show_color(page);
end

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
GUI.axes=axes('Units','pixels',...
      'PlotBoxAspectRatio',[1 1 1],...
      'Position',[50 30 400 400],...
      'Color',[0.98 0.98 0.98],...
      'Box','on', ...
      'XLim',[0 500],...
      'YLim',[0 500], ...
      'XColor','w','YColor','w',...
      'YDir','reverse', ...
      'Tag','picbagaxes',...
      'xtick',[],'ytick',[]);
hold on;
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    %function output_data(~,~)
        %if color_number>1
            %disp(color_list)
        %end
    %end
    function show_color(cur_page)
        len_list=size(color_list,1);
        for ii=(cur_page-1)*9+1:(cur_page-1)*9+9
            if(ii<=len_list)
                set(findobj('tag',num2str(ii-(cur_page-1)*9)),'backgroundcolor',color_list(ii,:)/255)
                set(findobj('tag',[num2str(ii-(cur_page-1)*9),'t']),'string',['[',num2str((color_list(ii,:)/255).*rgb_type+color_list(ii,:).*(~rgb_type)),']'])
            else
                set(findobj('tag',num2str(ii-(cur_page-1)*9)),'backgroundcolor',[1 1 1])
                set(findobj('tag',[num2str(ii-(cur_page-1)*9),'t']),'string','')
            end
        end  
    end

    function save_color(~,~)
        if ~isempty(get(GUI.text2,'string'))
            %set(findobj('tag',num2str(color_number)),'backgroundcolor',color/255)
            %set(findobj('tag',[num2str(color_number),'t']),'string',['[',num2str((color/255).*rgb_type+color.*(~rgb_type)),']'])
            color_list(color_number,:)=color;%(color/255).*rgb_type+color.*(~rgb_type);
            color_number=color_number+1;
            if color_number-1>(total_page-1)*9
                page=ceil(color_number/9);
                total_page=total_page+1;
                set(GUI.page,'string',[num2str(page),'/',num2str(total_page)]);
                
            end
            show_color(page)
        end
    end

    function delete_last(~,~)
        if color_number>=1
            %set(findobj('tag',num2str(color_number-1)),'backgroundcolor',[1 1 1])
            %set(findobj('tag',[num2str(color_number-1),'t']),'string','')
            color_list(end,:)=[];
            color_number=color_number-1;
            if color_number-2<=(total_page-2)*9
                page=ceil((color_number-1)/9);
                total_page=total_page-1;
                set(GUI.page,'string',[num2str(page),'/',num2str(total_page)]);
                
            end
            show_color(page)
        end
    end

    function delete_pic(~,~)
        control=0;
        set(findobj('Tag','picbagaxes'),...
        'XLim',[0 500],...
        'YLim',[0 500],...
        'Position',[50 30 400 400],...
        'Color',[0.98 0.98 0.98]);
        delete(a);
    end

    function clear_data(~,~)
        control=0;
        set(GUI.text1,'backgroundcolor',[1 1 1]);
        set(GUI.text2,'string','');
        set(findobj('Tag','picbagaxes'),...
        'XLim',[0 500],...
        'YLim',[0 500],...
        'Position',[50 30 400 400],...
        'Color',[0.98 0.98 0.98]);
        set(findobj('tag','recc'),'string','获取颜色');
        delete(a);
    end

    function get_color(~,~)
        if control==0
            set(GUI.text1,'backgroundcolor',[1 1 1]);
            set(GUI.text2,'string','');       
        end
        control=1;
        set(gcf,'WindowButtonMotionFcn',@whilemovefcn)  
        set(gcf,'WindowButtonDownFcn',@whileclickfcn)
    end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    function whilemovefcn(~,~)
        xy=get(gca,'CurrentPoint');
        x=xy(1,2);y=xy(1,1);
        if x<=x_limit&&y<=y_limit&&x>=0&&y>=0
            x(x>x_limit)=x_limit;
            y(y>y_limit)=y_limit;
            x(x<1)=1;
            y(y<1)=1;
            x=round(x);
            y=round(y);
            if control==1
                color=double([ima(x,y,1),ima(x,y,2),ima(x,y,3)]);
                set(GUI.text1,'backgroundcolor',color/255);
                set(GUI.text2,'string',['[',num2str((color/255).*rgb_type+color.*(~rgb_type)),']']);
            end
        else
            if control==1
                set(GUI.text1,'backgroundcolor',[1 1 1]);
                set(GUI.text2,'string','');    
            end
        end
    end

    function whileclickfcn(~,~)          
            xy=get(gca,'CurrentPoint');
            x=xy(1,2);y=xy(1,1);
            if x<=x_limit&&y<=y_limit&&x>=0&&y>=0
                control=0;
                set(GUI.text1,'backgroundcolor',color/255);
                set(GUI.text2,'string',['[',num2str((color/255).*rgb_type+color.*(~rgb_type)),']']);
                set(findobj('tag','recc'),'string','继续取色');
                %disp(color/255)
            end
    end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    function get_capture(~,~)
        screensize=get(0,'screensize');
        screensize=1.5*screensize;
        robot=java.awt.Robot();
        rectangle=java.awt.Rectangle();
        rectangle.x=0;
        rectangle.y=0;
        rectangle.width=screensize(3);
        rectangle.height=screensize(4);
        image=robot.createScreenCapture(rectangle);
        data=image.getData(); 
        temp=zeros(screensize(3)*screensize(4)*3,1);
        temp=data.getPixels(0,0,screensize(3),screensize(4),temp);  
        temp=uint8(temp);
        R=temp(1:3:end); 
        G=temp(2:3:end); 
        B=temp(3:3:end);
        R=reshape(R,[screensize(3),screensize(4)]); 
        G=reshape(G,[screensize(3),screensize(4)]); 
        B=reshape(B,[screensize(3),screensize(4)]);
        R=R';
        G=G';
        B=B';
        x_limit=screensize(4); 
        y_limit=screensize(3);
        leng=max([x_limit,y_limit]);
        set(findobj('Tag','picbagaxes'),...
        'XLim',[0 leng],...
        'YLim',[0 leng]);
        ima=cat(3,R,G,B);
        delete(a);
        a=imshow(ima);
    end
    function getImage(~,~)
        warning off;
        try
        [filename, pathname] = uigetfile({'*.jpg;*.tif;*.png;*.gif','All Image Files';...
            '*.*','All Files' });
        ima = imread([ pathname,filename]);
        [x,y,~]=size(ima);
        x_limit=x;y_limit=y;
        leng=max([x_limit,y_limit]);
        set(findobj('Tag','picbagaxes'),...
        'XLim',[0 leng],...
        'YLim',[0 leng]);
        delete(a);
        a=imshow(ima);
        catch
        end
    end  
end
相关标签: matlab