python中floor_Python torch.floor方法代码示例

本文整理汇总了Python中torch.floor方法的典型用法代码示例。如果您正苦于以下问题：Python torch.floor方法的具体用法？Python torch.floor怎么用？Python torch.floor使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在模块torch的用法示例。

在下文中一共展示了torch.floor方法的21个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Python代码示例。

示例1: map_roi_levels

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import floor [as 别名]

def map_roi_levels(self, rois, num_levels):

"""Map rois to corresponding feature levels by scales.

- scale < finest_scale * 2: level 0

- finest_scale * 2 <= scale < finest_scale * 4: level 1

- finest_scale * 4 <= scale < finest_scale * 8: level 2

- scale >= finest_scale * 8: level 3

Args:

rois (Tensor): Input RoIs, shape (k, 5).

num_levels (int): Total level number.

Returns:

Tensor: Level index (0-based) of each RoI, shape (k, )

"""

scale = torch.sqrt(

(rois[:, 3] - rois[:, 1]) * (rois[:, 4] - rois[:, 2]))

target_lvls = torch.floor(torch.log2(scale / self.finest_scale + 1e-6))

target_lvls = target_lvls.clamp(min=0, max=num_levels - 1).long()

return target_lvls

开发者ID:open-mmlab，项目名称:mmdetection，代码行数:22，

示例2: map_roi_levels

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import floor [as 别名]

def map_roi_levels(self, rois, num_levels):

"""Map rois to corresponding feature levels by scales.

- scale < finest_scale: level 0

- finest_scale <= scale < finest_scale * 2: level 1

- finest_scale * 2 <= scale < finest_scale * 4: level 2

- scale >= finest_scale * 4: level 3

Args:

rois (Tensor): Input RoIs, shape (k, 5).

num_levels (int): Total level number.

Returns:

Tensor: Level index (0-based) of each RoI, shape (k, )

"""

scale = torch.sqrt(

(rois[:, 3] - rois[:, 1] + 1) * (rois[:, 4] - rois[:, 2] + 1))

target_lvls = torch.floor(torch.log2(scale / self.finest_scale + 1e-6))

target_lvls = target_lvls.clamp(min=0, max=num_levels - 1).long()

return target_lvls

开发者ID:dingjiansw101，项目名称:AerialDetection，代码行数:22，

示例3: map_roi_levels

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import floor [as 别名]

def map_roi_levels(self, rois, num_levels):

"""Map rrois to corresponding feature levels by scales.

- scale < finest_scale: level 0

- finest_scale <= scale < finest_scale * 2: level 1

- finest_scale * 2 <= scale < finest_scale * 4: level 2

- scale >= finest_scale * 4: level 3

Args:

rois (Tensor): Input RRoIs, shape (k, 6). (index, x, y, w, h, angle)

num_levels (int): Total level number.

Returns:

Tensor: Level index (0-based) of each RoI, shape (k, )

"""

scale = torch.sqrt(rois[:, 3] * rois[:, 4])

target_lvls = torch.floor(torch.log2(scale / self.finest_scale + 1e-6))

target_lvls = target_lvls.clamp(min=0, max=num_levels - 1).long()

return target_lvls

开发者ID:dingjiansw101，项目名称:AerialDetection，代码行数:21，

示例4: fit_positive

​点赞 6

​

# 需要导入模块: import torch [as 别名]

# 或者: from torch import floor [as 别名]

def fit_positive(rows, cols, yx_min, yx_max, anchors):

device_id = anchors.get_device() if torch.cuda.is_available() else None

batch_size, num, _ = yx_min.size()

num_anchors, _ = anchors.size()

valid = torch.prod(yx_min < yx_max, -1)

center = (yx_min + yx_max) / 2

ij = torch.floor(center)

i, j = torch.unbind(ij.long(), -1)

index = i * cols + j

anchors2 = anchors / 2

iou_matrix = utils.iou.torch.iou_matrix((yx_min - center).view(-1, 2), (yx_max - center).view(-1, 2), -anchors2, anchors2).view(batch_size, -1, num_anchors)

iou, index_anchor = iou_matrix.max(-1)

_positive = []

cells = rows * cols

for valid, index, index_anchor in zip(torch.unbind(valid), torch.unbind(index), torch.unbind(index_anchor)):

index, index_anchor = (t[valid] for t in (index, index_anchor))

t = utils.ensure_device(to