pytorch使用中torch.uint8转换torch.float()卡住

已于 2023-05-09 22:40:31 修改
阅读量600 收藏
点赞数
分类专栏： 深度学习 文章标签： 深度学习 python 目标检测 pytorch
于 2023-05-09 22:39:40 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.youkuaiyun.com/gumeizhe/article/details/130590402
版权
在Python3.6.8环境下，使用yolov5目标检测项目时遇到torch.uint8类型数据转换为torch.float()时程序卡死的问题。原因可能是内存处理冲突。更新Python到最新版3.10解决了这个问题，但未测试其他Python版本，如3.7是否同样有效。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

pytorch使用中torch.uint8转换torch.float()卡住

问题描述：python3.6.8使用yolov5目标检测项目时，发现torch.uint8转换torch.float()卡住，即在代码
im = im.half() if self.model.fp16 else im.float()
这行卡死，也不报错，可能是torch在使用内存过程中起冲突了。

解决方案：将python升级到最新的3.10就完美解决。

注意：1、其它版本的python没试过，可能升级到3.7就可以了，但没试过。

报错处理：RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should...
cnjs1994的博客
04-11 3995
这篇博客解决的是pytorch训练图像分类模型中常常遇到的一个常见问题：就是模型在GPU,但是数据加载到了CPU
【Pytorch实用教程】Tensor的三大属性：torch.dtype、torch.device、torch.layout的详细讲解
若北辰
08-07 879
在PyTorch中，张量（Tensor）的三大属性是和。
pytorch---dtype=torch.uint8
m0_37347812的博客
04-12 1177
注意，这里最后会形成一个一维数组 当变为逻辑形式的时候会自动为unit8格式， 比如： 上面所述的： mask = x>7 或者 x!=7
图像处理中的uint8与float32之间的转换
星海燚燚的博客
03-03 1万+
# 预处理 def pre_process(raw_img, raw_mask, multiple): raw_mask = raw_mask.astype(np.float32) / 255. 上述代码目的是将uint8转换成float32 方法是先将图片转换成float32类型，再除以255，以得到0-1之前的数 import numpy as np im = im.astype(np.float32)/255 同理，float32转换成uint8 先将矩阵中每个数乘以255，再转换成ui
pytorc torch.uint8与torch.long/ torch. float
qq_29007291的博客
06-06 1万+
版本： pytorch1.0 目的： tensor中的index与mask 例子 下面针对torch.long 与torch.uint8数据类型在index/mask 中的不同作用进行分析 t = torch.rand(4，2) """ tensor([[0.5492, 0.2083], [0.3635, 0.5198], [0.8294, 0.9869], ...
PyTorch Float8 实验项目教程
gitblog_00805的博客
09-14 356
PyTorch Float8 实验项目教程 float8_experimental This repository contains the experimental PyTorch native float8 training UX ...
PyTorch学习之torch.nn.Conv2d函数
qq_46396470的博客
06-30 3417
是 PyTorch 中用于实现二维卷积层的类，这个类可以说是对的进一步封装，使其使用起来更加的傻瓜式。
PyTorch学习之torch.nn.functional.conv2d函数
qq_46396470的博客
06-29 1073
是 PyTorch 中用于进行二维卷积操作的函数。卷积操作是深度学习中卷积神经网络（CNN）的核心部分，用于提取图像特征，常见于图像分类、目标检测和语义分割等任务中。
解决pytorch-yolov3 train 报错的问题
12-20
UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instead https://github.com/eriklindernoren/PyTorch-YOLOv3/blob/master/models.py#L191 将model.py  obj_...
Input tensor should be a float tensor. Got torch.uint16.
最新发布
12-12
PyTorch中的大多数张量操作和模型输入都要求使用浮点类型（如`torch.float32`或`torch.float64`），而不是无符号整数类型（如`torch.uint16`）。 要解决这个问题，你可以将输入张量转换为浮点类型。以下是几种常见...
python将uint8类型的数组元素，转成float32类型，然后做除法的实现
weixin_41529093的博客
12-17 5327
仅作为记录，大佬请跳过。 使用.astype train_images数组是uint8类型的整数（如果直接除以255，则得到的数值会四舍五入的整数），博主想将其变成float，再进行除法这样得到的就是小数了。 不能使用.dtype 程序：（仅作为参照） train_images, train_labels, test_images, test_labels, train_nums, test_nums = a2svmministsample_gendatapropro.run() train_image
torch.int8和torch.uint8
weixin_56515806的博客
07-26 1096
torch.int8和torch.uint8都是PyTorch中的数据类型，它们的主要差别在于。
【torch张量】
m0_46256255的博客
03-19 643
参考：https://blog.csdn.net/xholes/article/details/81667211张量数据类型： 9种。
Pytorch学习笔记---2：张量
m0_64007201的博客
10-10 1801
对Pytorch张量的学习总结
(已解决)Ubuntu多显卡训练模型时程序卡死：torch torch 奈若何
行步至春深
09-26 3279
根因是cuda和pytorch的版本不匹配~经过这次最大的收获是理解了channel并学会了下载下安装包conda install --use-local 的安装方式
PyTorch张量操作详解
Maisule的博客
10-21 600
PyTorch建立在张量之上，PyTorch张量是一个n维数组，类似于NumPy数组。如果对NumPyNumpy数组PyTorch张量描述创建一个全 1 数组创建一个全 0 数组创建一个随机数组使用给定值创建数组x.shapex.shape获取数组形状在本节中，我们将学习如何定义和更改张量，将张量转换为数组，以及在计算设备之间移动张量。默认张量数据类型是，这是张量运算最常用的数据类型。1.print(x)复制代码2.print(x)复制代码。
torch中的数据类型和相互转换
程西西
12-12 2万+
文章目录1 torch.Tensor2 Data types3 Initializing and basic operations1）使用torch.tensor() 创建2）使用python list创建3）使用zeros ones函数创建4）通过torch.dtype指定类型， torch.device指定构造器5）零维张量6）torch.Tensor.item()从包含单个值的张量中获取Python数字4 type changes1）tensor间类型转换2）数据存储位置转换3）与Python数据类型
详解pytorch中的常见的Tensor数据类型以及类型转换
热门推荐
qq_43391414的博客
09-25 7万+
概览 本文主要讲pytorch中的常见的Tensor数据类型，例如：float32，float64，int32，int64。构造他们分别使用如下函数:torch.FloatTensor(),torch.DoubleTensor(),torch.IntTensor(),torch.LongTensor()。 Tensor的构建 1.32-bit floating point： a=torch.FloatTensor([1.0,3.0])#a=torch.Tensor([1.0,3.0])和前面等价 print
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值