复现centernet时，报错RuntimeError: CUDA error: out of memory

最新推荐文章于 2024-08-04 13:55:04 发布

itoshi rin

最新推荐文章于 2024-08-04 13:55:04 发布

阅读量178

点赞数 3

文章标签： python 目标检测机器学习 linux

本文链接：https://blog.csdn.net/sep21m_wyy/article/details/140192593

版权

运行

python test.py ctdet --dataset coco --exp_id coco_dla --load_model /root/CenterNet/exp/ctdet/coco_dla/model_last.pth --gpus 0 --test_scales 1

报错下面：

RuntimeError: CUDA error: out of memory

明明显存是够用的

在这里插入图片描述

解决办法：

找到自己虚拟环境的/root/miniconda3/lib/python3.6/site-packages/torch/nn/functional.py文件
在这里插入图片描述

在文件开头添加一句：

torch.backends.cudnn.enabled = False

在这里插入图片描述

在此运行：

CUDA_VISIBLE_DEVICES=0 python test.py ctdet --dataset coco --exp_id coco_dla --load_model /root/CenterNet/exp/ctdet/coco_dla/model_last.pth

正常进行测试：

在这里插入图片描述

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

itoshi rin

关注关注

3
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复现centernet时，报错RuntimeError: CUDA error: out of memory

复现centernet时，进行测试时报错RuntimeError: CUDA error: out of memory
复制链接

扫一扫

MATLAB算法实战应用案例精讲-【图像处理】小目标检测（附matlab代码实现）

qq_36130719的博客

01-13

4092

现有的小目标定义方式主要分为以下两类，即基于相对尺度的定义与基于绝对尺度的定义。(1)基于相对尺度即从目标与图像的相对比例这一角度考虑来对小目标进行定义。Chen[1]等对小目标做了如下定义：同一类别中所有目标实例的相对面积，即边界框面积与图像面积之比的中位数在0.08%~0.58%之间。除此以外，较为常见的还有以下几种：①目标边界框的宽高与图像的宽高比例小于一定值，较为通用的比例值为0.1；②目标边界框面积与图像面积的比值开方小于一定值，较为通用的值为0.03；

MATLAB算法实战应用案例精讲-【人工智能】语义分割（最终篇）(附实战应用案例及代码实现)

qq_36130719的博客

01-02

1584

SETR 是首个在分割任务上超越 CNN 的 Transformer 网络结构，该论文作者认为，虽然 FCN 及其衍生网络都取得了较好的成绩，但本质上都是 encoder-decoder 的结构，所以，该论文作者想要将分割任务构建成一个 sequence-to-sequence 的结构，所以就提出了一个纯 Transformer 的结构（无 CNN 和分辨率降低）。HRNet 通过将由高到低的分辨率特征结合起来的方式，实现对高分辨率特征的保留，并且不同分辨率的特征也进行了信息交换。

参与评论您还未登录，请先登录后发表或查看评论

CenterNet：Objects as Points论文学习笔记+代码复现(demo+训练数据)【检测部分】

weixin_38715903的博客

08-02

9044

目录 1.关键部分Heatmap了解 2.Centernet论文细节： 3.尝试复现CneterNet--INSTALL.md安装： 4.尝试复现CneterNet--跑跑demo.py： 5.尝试复现CneterNet--训练一下VOC： 1.关键部分Heatmap了解可以看懂的资料： 1.通俗地讲明白了heatmap的计算方法：http://bindog.gi...

旋转目标检测训练自己数据集+问题汇总

热门推荐

m0_46749624的博客

01-14

1万+

复现junjieliang大佬的旋转目标检测代码，将自己遇到的问题全部记录下来，包括数据集制作，训练与部署检测。

stable diffusion使用相关

hn_lgc的专栏

01-12

1000

IP Adapter比reference的效果要好，而且会快很多，适配于各种stable diffusion模型，还能和controlnet等一起用。IP Adapter是腾讯lab发布的一个新的Stable Diffusion适配器，它的作用是将你输入的图像作为图像提示词，本质上就像MJ的垫图。

OpenMMLab MMYOLO目标检测环境搭建（一）

qq_41627642的博客

09-11

2331

对于目标检测中的非开放世界数据集，训练和测试都是在一组固定的类上进行的，当应用于未经训练的类的图像时，有可能产生误报。论文指出，Mosaic+MixUp 可以大幅提升目标检测性能，但训练图片与自然图片的真实分布相去甚远，而且 Mosaic 大量的裁剪操作会带来很多不准确的标签框，因此，YOLOX 提出将去掉最后15个epoch的强增强，改用较弱的增强，这样检测器就可以避免不准确标记框的影响，在自然图片的数据分布下完成最终的收敛。首先，海报上的人与真人非常相似，而且人群密集的地方确实存在难以标注的人。

OSTrack 代码阅读记录

allrubots的博客

03-12

5786

实验记录

pixelNeRF代码流程

ysh的博客

05-07

3135

目录pixelNeRF: ReadMEEnvironment setupGetting the dataRunning the model (video generation)ShapeNet Multiple Categories (NMR)ShapeNet Single-Category (SRN)DTUReal Car ImagesOverview of flagsQuantitative evaluation instructions定量评估说明train.pyconfigdatasetnetcla

pytorch模型提示超出内存RuntimeError: CUDA out of memory.

01-20

跑模型时出现RuntimeError: CUDA out of memory.错误查阅了许多相关内容，原因是：GPU显存内存不够简单总结一下解决方法：将batch_size改小。取torch变量标量值时使用item()属性。可以在测试阶段添加如下代码：...

Vue项目报错：Uncaught SyntaxError: Unexpected token <

10-17

主要介绍了Vue项目报错：Uncaught SyntaxError: Unexpected token <，在引入第三方依赖的 JS 文件时，遇到的一个问题，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧

Python RuntimeError: thread.__init__() not called解决方法

09-22

然而，在使用`threading.Thread`进行多线程编程时，可能会遇到`RuntimeError: thread.__init__() not called`这样的错误。这个错误通常意味着你没有正确地初始化`threading.Thread`的实例。在标题和描述中提到的...

解决pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\\python3.6\\lib\\site-packa的问题

01-21

标题用pycharm导入numpy包的和使用时报错：RuntimeError: The current Numpy installation (‘D:\python3.6\...3.现在简单用numpy还是会报错：RuntimeError: The current Numpy installation (‘D:\python3.6\lib\site

【Python】pandas：排序、重复值、缺省值处理、合并、分组

yannan20190313的博客

07-30

1205

【Python】pandas：排序（sort_index,sort_values,nsmallest,nlargest）、重复值（duplicated,drop_duplicates,value_counts,nunique）、缺省值处理（isna,isnull,notna,notnull,fillna,dropna,replace）、合并（join,merge,append,concat）、分组（groupby）

本地部署 Llama-3-EvoVLM-JP-v2

最新发布

engchina的专栏

08-04

317

本地部署 Llama-3-EvoVLM-JP-v2

Selenium与WebDriver：Errno 8 Exec格式错误的多种解决方案

ip16yun的博客

07-30

453

在使用Selenium和WebDriver进行网页自动化时，可能会遇到各种错误。其中一个常见问题是执行格式错误（Errno 8 Exec format error）。这个错误通常在运行ChromeDriver时出现，错误提示涉及路径中的某个文件。本文将概述这个问题的背景，并提供多种解决方案，包括如何使用代理IP技术进行数据抓取。

十个好用到爆炸的Python技巧

qq_36807888的博客

07-30

673

生成器表达式与列表推导式类似，但它创建的是一个迭代器，而不是一个完整的列表。zip(names, ages) 创建一个迭代器，该迭代器生成由 names 和 ages 列表中对应位置的元素组成的元组。defaultdict(int) 创建一个字典，如果访问的键不存在，则默认值为 0 (int 类型的默认值)。惰性求值: 元素是在需要时才计算的，而不是预先计算好存储在内存中，这在处理包含复杂计算的序列时非常有用。wrapper 函数是装饰器函数内部定义的函数，它调用被装饰的函数，并将结果转换为大写。

python字符串数据容器练习（replace,split,count）

2301_79600015的博客

07-31

527

Python字符串是由一系列字符组成的数据类型。在Python中，字符串是不可变的，这意味着一旦创建了一个字符串，就不能修改它的内容。以下是关于Python字符串的一些重要知识点：1. 字符串的基本操作：拼接字符串：使用`+`运算符。重复字符串：使用`*`运算符。索引和切片：使用`[]`和`:`操作符访问字符串中的子串。长度：使用`len()`函数获取字符串长度。转换大小写：使用`upper()`和`lower()`方法。分割和连接：使用`split()`和`join()`方法。

Python面试题：如何使用WebSocket实现实时Web应用

bigorsmallorlarge的专栏

07-30

607

通过上述步骤，你可以使用 Django Channels 和 WebSocket 创建一个简单的实时 Web 应用。Django Channels 提供了强大的功能，使得在 Django 中实现实时功能变得更加容易。你可以在此基础上扩展，添加更多功能和复杂的逻辑。

yolov5报错RuntimeError: CUDA error: device-side assert triggered

09-14

这个错误是由于CUDA设备端断言触发引起的CUDA错误。根据给出的引用内容，主要原因可能是由于GPU显存不足导致的。解决这个问题的方法包括： 1. 将batch_size调小，减少每次传入模型的数据量。 2. 使用item()属性来获取torch变量的标量值，以减少显存的占用。 3. 在测试阶段，可以添加CUDA_LAUNCH_BLOCKING=1的代码，以确保CUDA错误能够被同步报告。另外，有人遇到类似问题时，发现错误并非是标签与分类的类别不一致，而是网络最后一层的代码有问题。需要确保网络输出的分类数与实际要求的分类数一致，例如将nn.Linear的输出尺寸改为正确的分类数。综上所述，你遇到的yolov5报错RuntimeError: CUDA error: device-side assert triggered可能是由于GPU显存不足所致。你可以尝试上述方法来解决这个问题。