没有合适的资源?快使用搜索试试~ 我知道了~
pytorch中的inference使用实例
2 下载量 26 浏览量
2020-12-20
13:57:38
上传
评论
收藏 41KB PDF 举报
温馨提示
试读
1页
这里inference两个程序的连接,如目标检测,可以利用一个程序提取候选框,然后把候选框输入到分类cnn网络中。 这里常需要进行一定的连接。 #加载训练好的分类CNN网络 model=torch.load('model.pkl') #假设proposal_img是我们提取的候选框,是需要输入到CNN网络的数据 #先定义transforms对输入cnn的网络数据进行处理,常包括resize、totensor等操作 data_transforms=transforms.Compose([transforms.RandomSizedCrop(224), transforms.ToTensor()
资源推荐
资源评论
资源评论
weixin_38567956
- 粉丝: 1
- 资源: 944
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功