用Lora驱动ID设计

最新推荐文章于 2024-10-02 10:53:34 发布

「已注销」

最新推荐文章于 2024-10-02 10:53:34 发布

阅读量139

点赞数

文章标签： python 开发语言

本文链接：https://blog.csdn.net/wutao22/article/details/130738094

版权

本文探讨了如何借助Lora技术进行ID设计，以脱毛仪为例，通过爬虫获取图片，使用Stable Diffusion WebUI进行预处理和训练。通过修改lora-scripts的参数并训练，最终生成初步模型，展现了将AI应用于ID设计的创新思路。

摘要由CSDN通过智能技术生成

相信现在大部分自研产品的公司都会有一个ID设计的职位，用来给产品设计符合需要的外观，这个外观是对一个硬件产品的外观造型、使用方式、人机交互进行设计的一个过程。它是一个产品的有形体现，能够实现产品与人高效、舒适、和谐的交互。ID设计是包含了一部分美学和艺术的东西在里面的技能，但凡艺术均需要灵感的激发，灵感这个东西飘忽不定，需要长期的经验积累以及艺术的熏陶，所以在灵感不足的时候，设计产品的就很容易陷入固定的审美中，导致产品的单一化，即使设计出来也讲不出故事。

在我前面的图文这是你幻想中的女友吗 (qq.com)有试用过Lora，LoRA的做法是，冻结预训练好的模型权重参数，然后在每个Transformer（Transforme就是GPT的那个T）块里注入可训练的层，由于不需要对模型的权重参数重新计算梯度，所以，大大减少了需要训练的计算量。基于这个思路，我们是否可以设计一个LORA来生成某类产品的ID呢。说干就干，我们就以现在热门的美容产品脱毛仪来进行测试。

首先我们需要准备一定数量的图片，图片从哪来呢？显然一张一张从网上去浏览保存不太现实，这个时候爬虫就派上了用场

import argparse,os,re,sys,urllib,json,socket,urllib.request,urllib.parse，urllib.error，time


timeout = 5
socket.setdefaulttimeout(timeout)


class Crawler:


    __time_sleep = 0.1
    __amount = 0
    __start_amount = 0
    __counter = 0
    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64; rv:23.0) Gecko/20100101 Firefox/23.0', 'Cookie': ''}
    __per_page = 30


    def __init__(self, t=0.1):
        self.time_sleep = t


    @staticmethod
    def get_suffix(name):
        m = re.search(r'\.[^\.]*$', name)
        if m.group(0) and len(m.group(0)) <= 5:
            return m.group(0)
        else:
            return '.jpeg'


    @staticmethod
    def handle_baidu_cookie(original_cookie, cookies):


        if not cookies:
            return original_cookie
        result = original_cookie
        for cookie in cookies:
            result += cookie.split(';')[0] + ';'
        result.rstrip(';')
        return result


    def save_image(self, rsp_data, word)