遗传算法求解几何问题

小白学视觉

共 6655字,需浏览 14分钟

 ·

2021-02-05 16:35

点击上方小白学视觉”,选择加"星标"或“置顶

重磅干货,第一时间送达

作者 | Victor Sim 

编译 | VK 

来源 | Towards Data Science

我最近看了一个关于“令人难以置信的直觉人工智能发明”的ted演讲:https://www.ted.com/talks/maurice_conti_the_incredible_inventions_of_intuitive_ai

这个ted演讲的特色是使用直观的人工智能生成汽车模型。这部分内容很简短,没有详细说明什么类型的人工智能以及它是如何实现的,所以我决定尝试用遗传算法复制这个项目的一个小规模版本。

我为什么选择遗传算法?与神经网络不同的是,遗传算法可以很容易地生成内容,而无需对图像进行卷积,然后将其转换回原始尺寸。但是,要找到正确格式的汽车模型数据是极其困难的,

概述了项目类型后,我应该如何简化问题?

概念

我将用简单的想法来代替制造低空气阻力的汽车,即创建一个由n个点组成的最大区域的形状,以点连接的方式。

形状的面积将使用鞋带(Shoelace )公式计算。从名称中可以推导出它是如何工作的:点坐标的交叉乘法创建了鞋带类型模式。

然后我将使用一种遗传算法(改编自这段代码:https://troysquillaci.me/simple-genetic-algorithm.html)来生成一组数字,然后将这些数字转换为坐标,从而绘制出一个形状。

代码

步骤1 |依赖项

import random
import numpy as np
from IPython.display import clear_outputdef sigmoid(x):
    return 1/(1+np.exp(-x))
    
def PolyArea(x,y):
    return 0.5*np.abs(np.dot(x,np.roll(y,1))-np.dot(y,np.roll(x,1)))

导入程序运行所需的基本依赖项。random用于随机生成智能体,numpy用于初始化和操作矩阵,IPython display用于清除屏幕上的混乱。

为了简单起见,我将在这个项目中使用的唯一激活函数是sigmoid函数。

polyarea函数是以numpy为数学基础的鞋带算法的实现。

步骤2 |实现类

class genetic_algorithm:
        
    def execute(pop_size,generations,threshold,network):
        class Agent:
            def __init__(self,network):
                class neural_network:
                    def __init__(self,network):
                        self.weights = []
                        self.activations = []
                        for layer in network:
                            if layer[0] != None:
                                input_size = layer[0]
                            else:
                                input_size = network[network.index(layer)-1][1]
                            output_size = layer[1]
                            activation = layer[2]
                            
                            self.weights.append(np.random.randn(input_size,output_size))
                            self.activations.append(activation)
                    def propagate(self,data):
                        input_data = data
                        for i in range(len(self.weights)):
                            z = np.dot(input_data,self.weights[i])
                            a = self.activations[i](z)
                            input_data = a
                        yhat = a
                        return yhat
                self.neural_network = neural_network(network)
                self.fitness = 0
                self.gene_drive = []
            def __str__(self):
                    return 'Loss: ' + str(self.fitness[0])

这是程序的开始,创建了遗传算法类和执行函数。

在agent的init中,初始化一个神经网络类,并根据给定的矩阵结构随机生成其权重。

步骤3 |创建种群

def generate_agents(population, network):
            return [Agent(network) for _ in range(population)]

该函数以种群大小和网络结构为参数,生成智能体的种群,神经网络随机生成权值。

步骤4 |计算适合度

def fitness(agents):
            for agent in agents:
                total_area = 0
                
                points = agent.neural_network.propagate(np.random.randn(1,semi_epochs))
                
                for shape in points:
                    x = list(shape[:num_points])
                    y = list(shape[num_points:])
                    
                    y.insert(0,0)
                    x.insert(0,0)
     y.insert(-1,0)
                    x.insert(-1,0)
     total_area += PolyArea(x,y) 
                    
                agent.fitness = total_area/semi_epochs
                
            return agents

我们将潜在点作为神经网络的输入。正因为如此,网络将进行多次尝试来生成形状,并记录这些形状的平均面积。

理论上,该算法将生成一个智能体,该智能体可以一致地生成具有n个点的高区域形状。观察这些形状可以帮助我们了解如何创建大面积的区域。

步骤5 |选择

def selection(agents):
            agents = sorted(agents, key=lambda agent: agent.fitness, reverse=True)
            print('\n'.join(map(str, agents)))
            agents = agents[:int(0.2 * len(agents))]
            return agents

程序的这一部分是选择算法,它根据智能体的适合度按逆序对它们进行排序。然后它会保留前五名。

步骤6 |交叉:

def crossover(agents,network,pop_size):
            offspring = []
            for _ in range((pop_size - len(agents)) // 2):
                parent1 = random.choice(agents)
                parent2 = random.choice(agents)
                child1 = Agent(network)
                child2 = Agent(network)
                
                shapes = [a.shape for a in parent1.neural_network.weights]
                
                genes1 = np.concatenate([a.flatten() for a in parent1.neural_network.weights])
                genes2 = np.concatenate([a.flatten() for a in parent2.neural_network.weights])
                
                split = random.randint(0,len(genes1)-1)child1_genes = np.array(genes1[0:split].tolist() + genes2[split:].tolist())
                child2_genes = np.array(genes1[0:split].tolist() + genes2[split:].tolist())
                
                for gene in parent1.gene_drive:
                    child1_genes[gene] = genes1[gene]
                    child2_genes[gene] = genes1[gene]
                    
                for gene in parent2.gene_drive:
                    child1_genes[gene] = genes2[gene]
                    child2_genes[gene] = genes2[gene]
                
                child1.neural_network.weights = unflatten(child1_genes,shapes)
                child2.neural_network.weights = unflatten(child2_genes,shapes)
                
                offspring.append(child1)
                offspring.append(child2)
            agents.extend(offspring)
            return agents

从种群的20%中随机选出两个父母。然后繁殖。如何做到这一点:

  1. 他们的权重平坦化(flatten)

  2. 找到一个随机的交点。这一点是单亲的遗传信息结束的地方,也是单亲遗传信息开始的地方。

  3. 将创建两个子代,然后将其添加到智能体列表中。这些子对象彼此不同,因为它们有不同的交点。

这有希望让好父母的优良品质遗传给后代。

步骤7 |突变

def mutation(agents):
            for agent in agents:
                if random.uniform(0.01.0) <= 0.1:
                    weights = agent.neural_network.weights
                    shapes = [a.shape for a in weights]flattened = np.concatenate([a.flatten() for a in weights])
                    randint = random.randint(0,len(flattened)-1)
                    flattened[randint] = np.random.randn()newarray = []
                    indeweights = 0
                    for shape in shapes:
                        size = np.product(shape)
                        newarray.append(flattened[indeweights : indeweights + size].reshape(shape
                        indeweights += size
                    agent.neural_network.weights = newarray
            return agents

有10%的几率发生突变。在这种情况下,变异指的是某个权重值被一个随机浮点值替换。通过将权重展平,找到要更改的随机权重。

步骤9 |执行

for i in range(generations):
            print('Generation',str(i),':')
            agents = generate_agents(pop_size,network)
            agents = fitness(agents)
            agents = selection(agents)
            agents = crossover(agents,network,pop_size)
            agents = mutation(agents)
            agents = fitness(agents)
            
            if any(agent.fitness > threshold for agent in agents):
                print('Threshold met at generation '+str(i)+' !')
                
            if i % 100:
                clear_output()
                
        return agents[0]

将最后一段代码粘贴到函数中,函数应该在调用时运行。

num_points = 3
semi_epochs = 100

network = [[semi_epochs,100,sigmoid],[None,num_points*2,sigmoid]]
ga = genetic_algorithm
agent = ga.execute(100,100,10,network)
weights = agent.neural_network.weights

我们可以改变程序可以用来创建形状的点的数量,以及程序可以生成点的次数,以得到平均值。改变这些值,你可能会发现一些有趣的东西!

原文链接:https://towardsdatascience.com/using-genetic-algorithms-to-solve-geometrical-problems-3789cb70bf25

下载1:OpenCV-Contrib扩展模块中文版教程
在「小白学视觉」公众号后台回复:扩展模块中文教程即可下载全网第一份OpenCV扩展模块教程中文版,涵盖扩展模块安装、SFM算法、立体视觉、目标跟踪、生物视觉、超分辨率处理等二十多章内容。

下载2:Python视觉实战项目31讲
小白学视觉公众号后台回复:Python视觉实战项目31讲即可下载包括图像分割、口罩检测、车道线检测、车辆计数、添加眼线、车牌识别、字符识别、情绪检测、文本内容提取、面部识别等31个视觉实战项目,助力快速学校计算机视觉。

下载3:OpenCV实战项目20讲
小白学视觉公众号后台回复:OpenCV实战项目20讲即可下载含有20个基于OpenCV实现20个实战项目,实现OpenCV学习进阶。

交流群


欢迎加入公众号读者群一起和同行交流,目前有SLAM、三维视觉、传感器自动驾驶、计算摄影、检测、分割、识别、医学影像、GAN算法竞赛等微信群(以后会逐渐细分),请扫描下面微信号加群,备注:”昵称+学校/公司+研究方向“,例如:”张三 + 上海交大 + 视觉SLAM“。请按照格式备注,否则不予通过。添加成功后会根据研究方向邀请进入相关微信群。请勿在群内发送广告,否则会请出群,谢谢理解~


浏览 28
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报