python - python函数参数的用例

Question

我有一个grids()在参数中使用关键字 self 命名的 python 函数，以及另一个名为build_targets(). 我想使用gridsizeand anchor_vecfromgrids()到另一个名为build_targets(). 这两个功能如下：

def grids(self, img_size=(608,608), gridsize=(19, 19), device='cpu', type=torch.float32):

    nx, ny = gridsize  # x and y grid size
    self.nx = nx
    self.ny = ny

    if isinstance(img_size, int):
        self.img_size = int(img_size)
    else:
        self.img_size = max(img_size)

    self.stride = self.img_size / max(gridsize)

    yv, xv = torch.meshgrid([torch.arange(ny), torch.arange(nx)])

    self.register_buffer('grid_xy', torch.stack((xv, yv), 2).view((1, 1, ny, nx, 2)).to(device)) 


    self.anchor_vec = self.anchors.to(device) / self.stride
    self.register_buffer('gridsize',torch.Tensor(gridsize).to(device))


def build_targets(model, targets):

    for i in layers_list:
        ng, anchor_vec = gridsize, anchor_vec # from grids() 

有什么方法可以使用gridsize和anchor_vecinbuild_targets()或者我如何在build_target()函数中调用它？

任何意见将不胜感激！

标签： python

Answer 1

似乎是类grids()中的方法，那么您有两种选择：

1. 修改build_targets()，通过添加self其参数使其成为类中的方法：

class SomeClass:
    def grids(...):
        self.gridsize = gridsize # (*)
        #some code

    def build_targets(self, model, targets):
        for i in layers_list:
            ng, anchor_vec = self.gridsize, self.anchor_vec # requires (*)
            # or
            ng, anchor_vec = (self.nx, self.ny), self.anchor_vec # does not requires (*)

来电：

some_instance.build_targets(some_model, some_targets)

2. 添加更多参数到build_targets()：

def build_targets(model, targets, gridsize, anchor_vec):
    for i in layers_list:
        ng, anchor_vec = gridsize, anchor_vec

来电：

build_targets(some_model, some_targets, (some_instance.nx, some_instance.ny), some_instance.anchor_vec)
# quite long line

通过这种方式，build_targets()对类是“独立的”。这样做相当于build_targets()搬出班级。