from d2l import torch as d2l img = d2l.plt.imread("D:/VOC/1.png") height, width = img.shape[:2] def display_anchors(fmap_width, fmap_height, s): d2l.set_figsize() fmap = d2l.zeros((1, 10, fmap_height, fmap_width)) anchors = d2l.multibox_prior(fmap, sizes=s, ratios=[1, 2, 0.5]) bbox_scale = d2l.tensor((width, height, width, height)) d2l.show_bboxes(d2l.plt.imshow(img).axes, anchors[0] * bbox_scale) display_anchors(fmap_width=4, fmap_height=4, s=[0.15])报错UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\TensorShape.cpp:3484.) return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]怎么修改
时间: 2024-04-09 19:31:14 浏览: 101
d2l-0.15.1-py3-none-any【d2l工具包-人工智能】
要解决这个警告,你可以使用`torch.meshgrid`函数,并显式传递索引参数。根据警告信息,索引参数是在`TensorShape.cpp`文件的第3484行触发的。
在你的代码中,你使用了库`d2l`,所以需要检查这个库是否提供了一个与警告相对应的函数,并且是否需要传递索引参数。
如果库`d2l`没有提供相应的函数,你可以尝试手动调用`torch.meshgrid`函数,并传递适当的索引参数。根据你的具体需求,索引参数的选择可能会有所不同。你可以查阅PyTorch文档来了解更多关于`torch.meshgrid`函数和索引参数的信息。
如果你无法确定如何传递索引参数,你可以尝试使用旧版本的PyTorch,因为在旧版本中可能不会触发这个警告。然而,这也意味着你可能无法使用最新版本的功能和修复程序。
请注意,这个警告是由PyTorch库内部触发的,因此我们无法直接控制或解决它。我们只能尝试根据警告信息和相关文档来调整代码,以适应即将发布的版本。
阅读全文