运行python程序时出现警告

原创

已于 2023-04-17 14:24:49 修改 · 1.9k 阅读

标签

#python

收录于

于 2022-04-13 17:13:47 首次发布

这篇博客详细记录了在运行Python程序时遇到的警告：使用torch.uint8类型进行索引已被弃用，建议改为torch.bool。作者通过定位问题，将`suppress`转换为bool类型，并解决了因使用[1-suppress]导致的运行时错误，改为使用[~suppress]。最后，展示了修复后的完整代码，确保了程序的正常运行。

运行python程序时出现警告：ndexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead

在定位到警告的语句后面添加.bool() warning：

定位到

 image_boxes.append(class_decoded_locs[1-suppress])
                image_labels.append(torch.LongTensor((1-suppress).sum().item() * [c]).to(device))
                image_scores.append(class_scores[1-suppress])

            # If no object in any class is found, store a placeholder for 'background'
            if len(image_boxes) == 0:
                image_boxes.append(torch.FloatTensor([[0., 0., 1., 1.]]).to(device))
                image_labels.append(torch.LongTensor([0]).to(device))
                image_scores.append(torch.FloatTensor([0.]).to(device))