14e451962036515076b17b5bc01e89282f62b681,models.py,,create_grids,#Any#Any#Any#Any#,236

Before Change

    self.stride = img_size / ng

    // build xy offsets
    grid_x = torch.arange(ng).repeat((ng, 1)).view((1, 1, ng, ng)).float()
    grid_y = grid_x.permute(0, 1, 3, 2)
    self.grid_xy = torch.stack((grid_x, grid_y), 4).to(device)

    // build wh gains
    self.anchor_vec = self.anchors.to(device) / self.stride

After Change

    self.stride = img_size / nx

    // build xy offsets
    yv, xv = torch.meshgrid([torch.arange(nx), torch.arange(ny)])
    self.grid_xy = torch.stack((xv, yv), 2).to(device).float().view((1, 1, nx, ny, 2))

    // build wh gains
    self.anchor_vec = self.anchors.to(device) / self.stride

Instances