优化算法matlab实现(二(补))框架增强:matlab动态图绘制

1.修改框架的目的

之前有不少问如下动态图是怎么绘制的?

这个是使用java编写的代码实现的,方式较为复杂,把每一代的位置画在一张图中,然后按顺序合成gif动态图。用java实现这种方式非常复杂,需要自己去实现绘制代码,然后保存成图片再合成。单绘制图像的代码就有几百行,而且有不少未解决的bug,比如一些位置的点只能用黑色或者白色才会在动态图中显示。
  将代码用matlab实现之后,发现使用matlab绘制动态图要比java容易一万倍,其效果如图。

所以在这里对优化算法框架进行一个小小的更新,让每个算法都能绘制群体位置动态图。

2.算法框架的修改

使用matlab绘制动态图的原理和使用java绘制的原理一样:即将每一代的所有个体的位置保存下来,然后每一代绘制成一张图,最后按顺序合成动态图。
  这里我们需要修改的部分由两点:
  1.保存所有个体的位置;
  2.绘制群体的位置。
  图像的显示以及动态图gif的生成,matlab已经帮我们完成,不需要单独去编码实现。
  需要修改的代码文件列表:
  优化算法matlab实现(二)框架编写中的框架。

文件名 描述
../optimization algorithm/frame/Unit.m 个体
../optimization algorithm/frame/Algorithm_Impl.m 算法主体

只需要修改这两个文件即可,不需要修改其他文件。
文件../optimization algorithm/frame/Unit.m中需要添加个体的历史位置及保存历史位置的方法,其实现如下:

% 个体基类
classdef Unit< handle
    properties
        % 个体的位置
        position;
        % 个体的适应度值
        value;
        % 记录所有的位置,画图用
        position_history_list;
    end
    
    methods
        function self = Unit()
        end
        
        % 将当前位置保存到position_history_list中
        function save(self)
            self.position_history_list=[self.position_history_list;self.position];
        end
    end
    
end

文件../optimization algorithm/frame/ Algorithm_Impl.m中每一代都需要调用上面的save方法来保存图像,其实现如下:

% 记录最优值
for i = 1:self.size
    if(self.unit_list(i).value>self.value_best)
        self.value_best = self.unit_list(i).value;
        self.position_best = self.unit_list(i).position;
    end
    % 保存每一代的位置
    self.unit_list(i).save();
end

文件../optimization algorithm/frame/ Algorithm_Impl.m中需要绘制群体位置的方法,其实现如下:

% 绘制动态图,2维图像:step为步长(几代绘制一张图像),is_save用来判断是否保存gif动态图
function draw2_gif(self,step,is_save,name)
    if self.dim < 2
        disp('维度太低,无法绘制图像');
        return
    end
    if step < 1
        step = 1;
    end
    f1 = figure;
    % 遍历每一代
    for i = 1:self.iter_max
        % 如果不满足步长,则跳过
        if mod(i,step) > 0 && i>1
            % 必须要绘制第一代,否则matlab会报错,原因未知
            continue
        end
        % 遍历每一个个体
        for s = 1:self.size
            cur_position = self.unit_list(s).position_history_list(i,:);
            scatter(cur_position(1),cur_position(2),10,'b','filled');
            hold on;
        end

        % 将文字绘制在左上角
        text(self.range_min_list(1),self.range_max_list(2),num2str(i),'FontSize',20);

        % 绘制显示区域
        range_size_x = self.range_max_list(1)-self.range_min_list(1);
        range_size_y = self.range_max_list(2)-self.range_min_list(2);
        axis([self.range_min_list(1)-0.2*range_size_x,self.range_max_list(1)+0.2*range_size_x, self.range_min_list(2)-0.2*range_size_y, self.range_max_list(2)+0.2*range_size_y]);

        axis equal;

        % 固定横纵坐标轴
        set(gca,'XLim',[self.range_min_list(1)-0.1*range_size_x self.range_max_list(1)+0.1*range_size_x]);
        set(gca,'YLim',[self.range_min_list(2)-0.1*range_size_y self.range_max_list(2)+0.1*range_size_y]);
        % 每0.1绘制一次
        pause = 0.1;
        % 需要保存git则设置is_save = true
        if is_save
            %下面是保存为GIF的程序
            frame=getframe(gcf);
            % 返回单帧颜色图像
            imind=frame2im(frame);
            % 颜色转换
            [imind,cm] = rgb2ind(imind,256);
            filename = [name,'_2d.gif'];
            if i==1
                 imwrite(imind,cm,filename,'gif', 'Loopcount',inf,'DelayTime',1e-2);
            else
                 imwrite(imind,cm,filename,'gif','WriteMode','append','DelayTime',pause);
            end
        end
        % 绘制完就清除,绘制下一代
        clf;
    end
    % 绘制完成关闭窗口
    close(f1);    
end
        

上面绘制的是2维动态图,我们可以将其类推至3维,来绘制3维动态图。当然图像最多只能绘制到3维,毕竟我们生活在3维空间,只能直观看到3维。

% 绘制动态图,3维图像:step为步长(几代绘制一张图像),is_save用来判断是否保存gif动态图
function draw3_gif(self,step,is_save,name)
    if self.dim < 3
        disp('维度太低,无法绘制三维图像');
        return
    end
    if step < 1
        step = 1;
    end
    f1 = figure;
    % 遍历每一代
    for i = 1:self.iter_max
        % 如果不满足步长,则跳过
        if mod(i,step) > 0 && i>1
            % 必须要绘制第一代,否则matlab会报错,原因未知
            continue
        end
        % 遍历每一个个体
        for s = 1:self.size
            cur_position = self.unit_list(s).position_history_list(i,:);
            scatter3(cur_position(1),cur_position(2),cur_position(3),10,'b','filled');
            hold on;
        end

        % 将文字绘制在左上角
        text(self.range_min_list(1),self.range_max_list(2),self.range_max_list(3),num2str(i),'FontSize',20);

        % 绘制显示区域
        range_size_x = self.range_max_list(1)-self.range_min_list(1);
        range_size_y = self.range_max_list(2)-self.range_min_list(2);
        range_size_z = self.range_max_list(3)-self.range_min_list(3);
        axis([self.range_min_list(1)-0.2*range_size_x,self.range_max_list(1)+0.2*range_size_x, self.range_min_list(2)-0.2*range_size_y, self.range_max_list(2)+0.2*range_size_y,self.range_min_list(3)-0.2*range_size_z, self.range_max_list(3)+0.2*range_size_z]);

        axis equal;

        % 固定横纵坐标轴
        set(gca,'XLim',[self.range_min_list(1)-0.1*range_size_x self.range_max_list(1)+0.1*range_size_x]);
        set(gca,'YLim',[self.range_min_list(2)-0.1*range_size_y self.range_max_list(2)+0.1*range_size_y]);
        set(gca,'ZLim',[self.range_min_list(3)-0.1*range_size_z self.range_max_list(3)+0.1*range_size_z])
        % 每0.1绘制一次
        pause = 0.1;
        % 需要保存git则设置is_save = true
        if is_save
            %下面是保存为GIF的程序
            frame=getframe(gcf);
            % 返回单帧颜色图像
            imind=frame2im(frame);
            % 颜色转换
            [imind,cm] = rgb2ind(imind,256);
            filename = [name,'_3d.gif'];
            if i==1
                 imwrite(imind,cm,filename,'gif', 'Loopcount',inf,'DelayTime',1e-2);
            else
                 imwrite(imind,cm,filename,'gif','WriteMode','append','DelayTime',pause);
            end
        end
        % 绘制完就清除,绘制下一代
        clf;
    end
    % 绘制完成关闭窗口
    close(f1);    
end

3维图像效果如图:

3.完整框架代码

总目录:../optimization algorithm
框架目录:../optimization algorithm/frame
框架文件:

文件名 描述
../optimization algorithm/frame/Unit.m 个体
../optimization algorithm/frame/Algorithm_Impl.m 算法主体

文件内容:
Unit.m

% 个体基类
classdef Unit< handle
    properties
        % 个体的位置
        position;
        % 个体的适应度值
        value;
        % 记录所有的位置,画图用
        position_history_list;
    end
    
    methods
        function self = Unit()
        end
        
        % 将当前位置保存到position_history_list中
        function save(self)
            self.position_history_list=[self.position_history_list;self.position];
        end
    end
    
end

Algorithm_Impl.m

% 优化算法基类
classdef Algorithm_Impl < handle
    properties

        %当前最优位置
        position_best;
        %当前最优适应度
        value_best;
        %历史最优适应度
        value_best_history;
        %历史最优位置
        position_best_history;
        %是否为求最大值,默认为是
        is_cal_max;
        %适应度函数,需要单独传入
        fitfunction;
        % 调用适应度函数次数
        cal_fit_num = 0;
    end
    properties(Access = protected)
        %维度
        dim;
        %种群中个体的数量
        size;
        %最大迭代次数
        iter_max;
        %解空间下界
        range_min_list;
        %解空间上界
        range_max_list;
        %种群列表
        unit_list;
    end
    
    methods
         % 运行,调用入口
        function run(self)
            tic
            self.init()
            self.iteration()
            toc
            disp(['运行时间: ',num2str(toc)]);

        end
        
        % 绘制动态图,2维图像
        function draw2_gif(self,step,is_save,name)
            if self.dim < 2
                disp('维度太低,无法绘制图像');
                return
            end
            if step < 1
                step = 1;
            end
            f1 = figure;
            % 遍历每一代
            for i = 1:self.iter_max
                % 如果不满足步长,则跳过
                if mod(i,step) > 0 && i>1
                    % 必须要绘制第一代,否则matlab会报错,原因未知
                    continue
                end
                % 遍历每一个个体
                for s = 1:self.size
                    cur_position = self.unit_list(s).position_history_list(i,:);
                    scatter(cur_position(1),cur_position(2),10,'b','filled');
                    hold on;
                end

                % 将文字绘制在左上角
                text(self.range_min_list(1),self.range_max_list(2),num2str(i),'FontSize',20);

                % 绘制显示区域
                range_size_x = self.range_max_list(1)-self.range_min_list(1);
                range_size_y = self.range_max_list(2)-self.range_min_list(2);
                axis([self.range_min_list(1)-0.2*range_size_x,self.range_max_list(1)+0.2*range_size_x, self.range_min_list(2)-0.2*range_size_y, self.range_max_list(2)+0.2*range_size_y]);

                axis equal;

                % 固定横纵坐标轴
                set(gca,'XLim',[self.range_min_list(1)-0.1*range_size_x self.range_max_list(1)+0.1*range_size_x]);
                set(gca,'YLim',[self.range_min_list(2)-0.1*range_size_y self.range_max_list(2)+0.1*range_size_y]);
                % 每0.1绘制一次
                pause = 0.1;
                % 需要保存git则设置is_save = true
                if is_save
                    %下面是保存为GIF的程序
                    frame=getframe(gcf);
                    % 返回单帧颜色图像
                    imind=frame2im(frame);
                    % 颜色转换
                    [imind,cm] = rgb2ind(imind,256);
                    filename = [name,'_2d.gif'];
                    if i==1
                         imwrite(imind,cm,filename,'gif', 'Loopcount',inf,'DelayTime',1e-2);
                    else
                         imwrite(imind,cm,filename,'gif','WriteMode','append','DelayTime',pause);
                    end
                end
                % 绘制完就清除,绘制下一代
                clf;
            end
            % 绘制完成关闭窗口
            close(f1);    
        end
        
        % 绘制动态图,3维图像
        function draw3_gif(self,step,is_save,name)
            if self.dim < 3
                disp('维度太低,无法绘制三维图像');
                return
            end
            if step < 1
                step = 1;
            end
            f1 = figure;
            % 遍历每一代
            for i = 1:self.iter_max
                % 如果不满足步长,则跳过
                if mod(i,step) > 0 && i>1
                    % 必须要绘制第一代,否则matlab会报错,原因未知
                    continue
                end
                % 遍历每一个个体
                for s = 1:self.size
                    cur_position = self.unit_list(s).position_history_list(i,:);
                    scatter3(cur_position(1),cur_position(2),cur_position(3),10,'b','filled');
                    hold on;
                end

                % 将文字绘制在左上角
                text(self.range_min_list(1),self.range_max_list(2),self.range_max_list(3),num2str(i),'FontSize',20);

                % 绘制显示区域
                range_size_x = self.range_max_list(1)-self.range_min_list(1);
                range_size_y = self.range_max_list(2)-self.range_min_list(2);
                range_size_z = self.range_max_list(3)-self.range_min_list(3);
                axis([self.range_min_list(1)-0.2*range_size_x,self.range_max_list(1)+0.2*range_size_x, self.range_min_list(2)-0.2*range_size_y, self.range_max_list(2)+0.2*range_size_y,self.range_min_list(3)-0.2*range_size_z, self.range_max_list(3)+0.2*range_size_z]);

                axis equal;

                % 固定横纵坐标轴
                set(gca,'XLim',[self.range_min_list(1)-0.1*range_size_x self.range_max_list(1)+0.1*range_size_x]);
                set(gca,'YLim',[self.range_min_list(2)-0.1*range_size_y self.range_max_list(2)+0.1*range_size_y]);
                set(gca,'ZLim',[self.range_min_list(3)-0.1*range_size_z self.range_max_list(3)+0.1*range_size_z])
                % 每0.1绘制一次
                pause = 0.1;
                % 需要保存git则设置is_save = true
                if is_save
                    %下面是保存为GIF的程序
                    frame=getframe(gcf);
                    % 返回单帧颜色图像
                    imind=frame2im(frame);
                    % 颜色转换
                    [imind,cm] = rgb2ind(imind,256);
                    filename = [name,'_3d.gif'];
                    if i==1
                         imwrite(imind,cm,filename,'gif', 'Loopcount',inf,'DelayTime',1e-2);
                    else
                         imwrite(imind,cm,filename,'gif','WriteMode','append','DelayTime',pause);
                    end
                end
                % 绘制完就清除,绘制下一代
                clf;
            end
            % 绘制完成关闭窗口
            close(f1);    
        end
    end
    
    methods (Access = protected)
        % 构造函数
        function self = Algorithm_Impl(dim,size,iter_max,range_min_list,range_max_list)
            self.dim =dim;
            self.size = size;
            self.iter_max = iter_max;
            self.range_min_list = range_min_list;
            self.range_max_list = range_max_list;
            %默认为求最大值
            self.is_cal_max = true;
        end
        
        % 初始化
        function init(self)
            self.position_best=zeros(1,self.dim);
            self.value_best_history=[];
            self.position_best_history=[];
            %设置初始最优值,由于是求最大值,所以设置了最大浮点数的负值
            self.value_best = -realmax('double');
        end
        
        % 开始迭代
        function iteration(self)
            for iter = 1:self.iter_max
                self.update(iter)
            end
        end
        
        % 处理一次迭代
        function update(self,iter)
            % 记录最优值
            for i = 1:self.size
                if(self.unit_list(i).value>self.value_best)
                    self.value_best = self.unit_list(i).value;
                    self.position_best = self.unit_list(i).position;
                end
                % 保存每一代的位置
                self.unit_list(i).save();
            end
            disp(['第' num2str(iter) '代']);
            if(self.is_cal_max)
                self.value_best_history(end+1) = self.value_best;
                disp(['最优值=' num2str(self.value_best)]);
            else
                self.value_best_history(end+1) = -self.value_best;
                disp(['最优值=' num2str(-self.value_best)]);
            end
            self.position_best_history = [self.position_best_history;self.position_best];
            disp(['最优解=' num2str(self.position_best)]);
        end
        
        function value = cal_fitfunction(self,position)
            if(isempty(self.fitfunction))
                value = 0;
            else
                % 如果适应度函数不为空则返回适应度值
                if(self.is_cal_max)
                    value = self.fitfunction(position);
                else
                    value = -self.fitfunction(position);
                end
            end
            self.cal_fit_num = self.cal_fit_num+1;
        end
        
        % 越界检查,超出边界则停留在边界上
        function s=get_out_bound_value(self,position,min_list,max_list)
          if(~exist('min_list','var'))
              min_list = self.range_min_list;
          end
          if(~exist('max_list','var'))
              max_list = self.range_max_list;
          end
          % Apply the lower bound vector
          position_tmp=position;
          I=position_tmpmax_list;
          position_tmp(J)=max_list(J);
          % Update this new move
          s=position_tmp;
        end
        
        % 越界检查,超出边界则在解空间内随机初始化
        function s=get_out_bound_value_rand(self,position,min_list,max_list)
          if(~exist('min_list','var'))
              min_list = self.range_min_list;
          end
          if(~exist('max_list','var'))
              max_list = self.range_max_list;
          end
          position_rand = unifrnd(self.range_min_list,self.range_max_list);
          % Apply the lower bound vector
          position_tmp=position;
          I=position_tmpmax_list;
          position_tmp(J)=position_rand(J);
          % Update this new move
          s=position_tmp;
        end
        
        
    end

    events
    end
end

4.测试代码

使用之前的差分进化算法测试一下动态图的绘制(需要实现优化算法matlab实现(七)差分进化算法matlab实现中的相关代码)。

注意调用方法是需要自行决定步长,即draw2_gif()和draw3_gif中的step参数。步长越小绘制的图片越多,执行的越慢。

测试F1

文件名:../optimization algorithm/algorithm_differential_evolution/Test.m

%% 清理之前的数据
% 清除所有数据
clear all;
% 清除窗口输出
clc;

%% 添加目录
% 将上级目录中的frame文件夹加入路径
addpath('../frame')


%% 选择测试函数
Function_name='F1';
%[最小值,最大值,维度,测试函数]
[lb,ub,dim,fobj]=Get_Functions_details(Function_name);

%% 算法实例
% 种群数量
size = 50;
% 最大迭代次数
iter_max = 1000;
% 取值范围上界
range_max_list = ones(1,dim)*ub;
% 取值范围下界
range_min_list = ones(1,dim)*lb;

% 实例化差分进化算法类
base = DE_Base(dim,size,iter_max,range_min_list,range_max_list);
base.is_cal_max = false;
% 确定适应度函数
base.fitfunction = fobj;
% 运行
base.run();
disp(base.cal_fit_num);

%% 绘制2维图像,每10代绘制一次,且保存gif图像到本地
%base.draw2_gif(10,true,base.name);

% 绘制3维图像,每10代绘制一次,且保存gif图像到本地
base.draw3_gif(10,true,base.name);

%% 绘制图像
figure('Position',[500 500 660 290])
%Draw search space
subplot(1,2,1);
func_plot(Function_name);
title('Parameter space')
xlabel('x_1');
ylabel('x_2');
zlabel([Function_name,'( x_1 , x_2 )'])
%Draw objective space
subplot(1,2,2);
% 绘制曲线,由于算法是求最大值,适应度函数为求最小值,故乘了-1,此时去掉-1
semilogy((base.value_best_history),'Color','r')
title('Objective space')
xlabel('Iteration');
ylabel('Best score obtained so far');
% 将坐标轴调整为紧凑型
axis tight
% 添加网格
grid on
% 四边都显示刻度
box off
legend(base.name)
display(['The best solution obtained by ',base.name ,' is ', num2str(base.value_best)]);
display(['The best optimal value of the objective funciton found by ',base.name ,' is ', num2str(base.position_best)]);
步长为10,2d图
步长为10,3d图

版权声明:
作者:ht
链接:https://www.techfm.club/p/45428.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>