请选择 进入手机版 | 继续访问电脑版

Hello Mat

 找回密码
 立即注册

QQ登录

只需一步,快速开始

查看: 576|回复: 1

tensor和numpy、List基本转换

[复制链接]

41

主题

67

帖子

1152

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
1152
发表于 2020-5-17 17:35:19 | 显示全部楼层 |阅读模式
tensor和numpy、List基本转换:
(1)cuda--cpu、tensor--numpy
  1. pred_mask = (outputs*255).cpu().data.numpy().astype(np.uint8)
复制代码
(2)tensor大小和类型
  1. print(outputs.type())
  2. print(outputs.size())
复制代码
(3)numpy大小和类型
  1. type(pred_mask)
  2. print(pred_mask.shape)
复制代码











回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Python|Opencv|MATLAB|Halcom.cn  

GMT+8, 2020-6-2 13:21 , Processed in 0.153246 second(s), 27 queries .

Powered by Discuz! X3.2

© 2001-2013 Comsenz Inc.

快速回复 返回顶部 返回列表