pytorch torchvision.ops.roi_align

 最后发现pytorch的roialign不是和网上说的一样啊，他输入的bbox框坐标是需要相对于input的坐标的啊！比如inpu的featuremap的width是24，那么就要求框坐标是[0-23]之间的数！

下面要注意bbox的位置数值应该是大于1的数字，小数都是错的

import torchvision
import torch

input_ = torch.rand(3, 128, 24, 96) #[b,c,h,w]

bbox = [[0,0.1,0.15,0.4,0.6],
        [0,0.1,0.15,0.4,0.6],
[1,0.01,0.15,0.4,0.6],
[1,0.1,0.25,0.7,0.6],
[2,0.01,0.14,0.5,0.6],
[2,0.1,0.45,0.4,0.6],
[2,0.01,0.45,0.12,0.6],
[0,0.01,0.25,0.15,0.6]]

bbox_1 = torch.tensor(bbox)#[8,5]

#[8,128,7,7]
roi = torchvision.ops.roi_align(input=input_, boxes=bbox_1,
                                        output_size=(7, 7))
# pytorch # 深度学习 # python
Logo
乐奇 Rokid 开放社区

更多推荐

KOReader 安装教程：电子墨水屏上重排 PDF，免费电子书阅读器

在电子墨水屏上读 PDF，要么放大拖拽，要么只能看原版的小字。KOReader 是一款面向电子墨水屏的开源、免费电子书阅读器，支持 PDF、EPUB、DjVu、FB2 等十几种格式，还能把固定版式的 PDF 重排成适合小屏的排版。## 📌 这类人值得装- **啃 PDF 论文的人**：学术论文、技术文档、扫描版书籍重排之后，文字按屏幕宽度排，不用再在放大和翻页之间来回切。- **多设备

avatar 乐奇 Rokid 开放社区

Hyperswitch config_importer 实战：将 TOML 配置文件一键转换为 Kubernetes 环境变量

本文以 Hyperswitch 仓库中的 `config_importer` 工具 crate 为核心，完整讲解它如何把 Hyperswitch 的 TOML 配置文件（如 `config/development.toml`、`config/deployments/drainer.toml`）转换为环境变量的键值对，并导出为 Kubernetes 可用的 JSON 数组格式。读完本文，你将掌握该工

avatar 乐奇 Rokid 开放社区

AI眼镜微型件连接 激光焊和胶接怎么选

所谓智能眼镜微型件连接工艺选型，是指在镜腿铰链、中梁、摄像头模组、FPC 与 PCB 互连、腔体密封、塑料壳体这些尺度在亚毫米到数毫米之间的连接界面上，按热预算、接头强度、气密要求与疲劳寿命四项约束，在激光熔焊、激光锡焊、回流焊、电阻焊、结构胶接与超声波焊之间做出取舍的工艺决策方法（来源：精密焊接在智能眼镜制造中的应用行业资料）。

avatar 乐奇 Rokid 开放社区