动手学深度学习
LearnerzzZ
这个作者很懒,什么都没留下…
展开
-
【动手学深度学习】之非极大值抑制(NMS)代码实现
import torchfrom d2l import torch as d2l# 更改打印设置torch.set_printoptions(2)def show_bboxes(axes, bboxes, labels=None, colors=None): """显⽰所有边界框""" def _make_list(obj, default_values=None): if obj is None: obj = default_val.原创 2021-12-14 20:53:21 · 3007 阅读 · 0 评论 -
【动手学深度学习】之锚框生成函数代码实现
【代码】:import torchfrom d2l import torch as d2l# 更改打印设置torch.set_printoptions(2)def multibox_prior(data, sizes, ratios): """生成以每个像素为中心具有不同形状的锚框。""" # 图片的高和宽 in_height, in_width = data.shape[-2:] # 查看操作设备(CPU/GPU),与锚框的参数 device, n原创 2021-12-12 10:46:44 · 3315 阅读 · 0 评论