【啥都生】分类项目中的模型搭建代码解析

news2024/11/18 15:33:06

def build_model(cfg):
    if isinstance(cfg, list):
        modules = [
            eval(cfg_.pop("type"))(**cfg_) for cfg_ in cfg
        ]
        return Sequential(*modules)
    else:
        return eval(cfg.pop("type"))(**cfg)

b站up啥都生维护的分类项目
这段代码的功能是完成模型搭建，其中有三个python的巧妙用法，这里是自己的理解，有不对的地方希望各路大佬指出！

①eval语句（将参数变成函数返回）
②初始化__init__文件的作用（调包列表）
③*和**解包语法（*是将列表解包为单个元素**是解包字典）
isinstance和sequential

1.eval 语句

这个函数的功能是将参数变为可执行语句。
例子：
例子1

2.`**`语法

2.1 单个`*`

2.1.1 单个*表示乘和倍数(用在两个变量中间)

print(3*4)         # 输出：12
print('nihao '*3)  # 输出：nihao nihao nihao

2.1.2 单个*表示组合、解包元组（用在单个变量前）

# 组合
# 接受任意多个参数放在一个元组内   
def demo(*p):
    print(p)


demo((1,2,3))      # 输出：((1, 2, 3),)
demo(1,2,3)        # 输出：(1, 2, 3)

# 解包
# 函数串参时将一个参数解包为三个
def demo(a, b, c):
    print(a, b, c)
a1 = ['ni', 'hao', 'a']
b1 = ['da', 'tai', 'yang']
c1 = ['shu', 'ya', 'zi']
demo(a1,b1,c1)  # 输出： ['ni', 'hao', 'a'] ['da', 'tai', 'yang'] ['shu', 'ya', 'zi']
# 这句中的*语法就是将a数组的三个元素解包，变为三个字符串分别传给形参a,b,c
demo(*a1)       # 输出： ni hao a 

# 解包为独立的元素输出
a = [1,2,3,4]
print(*a, sep='--')  # 输出：1--2--3--4

2.2 双星号`**`

2.2.1 双星号**表示取幂（用在两个变量之间）

print(7**2)  # 输出：49

2.2.2双星号**表示组合、解包字典（用在单个变量前）

# 组合
def demo(**p):
    return p
print(demo(x=1,y=2,z=6,q=9)) #输出：{'x': 1, 'y': 2, 'z': 6, 'q': 9}

# 解包
dic = {'x':'1','y':'2','z':'6'}
for s in dic:
    print(dic.get(s))
st = "{z}---{z}---{z} ".format(**dic)
st2 = '{z}---{x}---{x}---{y}'.format(**dic)
print(st, st2)           #输出：6---6---6  6---1---1---2

3.init初始化文件

导包的时候from某某import **号表示导入该文件夹下所有包，那是如何精确导入呢？
通过在文件夹下定义__init__.py文件:把该文件夹视作一个模块，当某程序调用该文件夹功能是率先访问__init__.py文件
当我们使用from某某import *导包时，首先访问的是__init__.py文件中的__all__列表，该列表中的元素就是按顺序对应前面导入的模型。
在这里插入图片描述

4 在代码中的解析

首先判断传入参数是不是列表

isinstance
作用：判断是否是同一类型，返回true or false
在这段代码中就是判断cfg是不是列表

isinstance(object, classinfo)
object -- 实例对象。
classinfo -- 可以是直接或间接类名、基本类型或者由它们组成的元组
isinstance() 与 type() 区别：

type() 不会认为子类是一种父类类型，不考虑继承关系。
isinstance() 会认为子类是一种父类类型，考虑继承关系。
如果要判断两个类型是否相同推荐使用 isinstance()。

如果是列表
代码使用列表推导式遍历cfg列表，将其中的每个字典元素cfg_处理成一个模块实例，并将所有模块组成一个modules列表。列表中的每个字典元素表示一个模块的配置，其中"type"键对应的值是模块的类名（字符串形式），其他键值对是模块的初始化参数。
eval(cfg_.pop(“type”))这一部分是为了根据字符串形式的模块类名创建模块实例。eval()函数将字符串作为表达式进行求值，从而创建相应的模块实例。pop(“type”)用于从cfg_字典中弹出键为"type"的元素，以便接下来的**cfg_只包含模块的初始化参数。
最后，使用Sequential(*modules)将所有的模块组成一个Sequential模型，并将其作为函数的返回值
如果不是列表
这里的处理方式与之前类似，但是不同之处在于这里的cfg表示一个模块的配置字典，而不是多个模块的配置列表。代码首先使用eval()根据字符串形式的模块类名创建模块实例，然后将配置参数**cfg传递给模块的初始化函数来创建模块实例，并将其作为函数的返回值。
请注意，由于代码使用了eval()函数，这可能会导致潜在的安全风险，特别是如果cfg的内容来自不受信任的来源。如果可能的话，最好使用其他方法来创建模块实例，比如通过模块类名的映射字典，以避免eval()的使用。
总结起来，这段代码是一个根据传入配置cfg构建神经网络模型的函数，如果cfg是列表，则构建一个由多个模块组成的Sequential模型，如果cfg是字典，则构建单个模块。在创建模块时，cfg中的"type"键对应的字符串表示模块类名，其他键值对表示模块的初始化参数。

例如cfg是model_cfg的实例化对象如下所示
在这里插入图片描述