训练CenterNet遇到的问题

最新推荐文章于 2024-05-15 10:53:24 发布
最新推荐文章于 2024-05-15 10:53:24 发布
阅读量509 收藏
点赞数
分类专栏： 目标检测 文章标签： 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/TYtangyan/article/details/108704260
版权

最近跑centerNet的模型,但是在DCN部分遇到错误,记录一下

配置:ubuntu18.04

cuda10.0

pytorch:1.10

网络:  https://github.com/xingyizhou/CenterNet

问题1:编译自带的DCNv2的时候./make.sh

ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.

自带的这个dcnv不太好用,需要使用下面这个进行编译

https://github.com/CharlesShang/DCNv2

这个问题网上问的也比较多
 

最后我的解决方式为pytorch1.1.0变为pytorch1.3.1模型可以成功运行DCN的网络部分.

在make以后测试的时候遇到以下输出,不用管它,运行centerNet的网络,可以正常运行

版本:

torch                  1.3.1                  
torchvision            0.4.2

新生代农民工！
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
CenterNet配置及问题详解
Sig的博客
09-29 8490
作者原版github：https://github.com/xingyizhou/CenterNet Install 按照readme文件夹中的INSTALL.md操作： 0.创建一个虚拟环境 conda create --name CenterNet python=3.6#创建一个名为CenterNet的python3.6虚拟环境 source activate CenterNet #激活...
【已解决】关于Centernet复现，编译最新DCNv2出现的问题 LINK : fatal error LNK1181: 无法打开输入文件
大道至简
04-26 4702
错误描述： error: command 'D:\\vs2017\\VC\\Tools\\MSVC\\14.16.27023\\bin\\HostX86\\x64\\link.exe' failed with exit status 1181 执行命令： python setup.py build develop 最新DCNv2下载地址：（支持pytorch1.8） [https://codechina.csdn.net/mirrors/jinfagang/DCNv2_latest](https://c
CenterNet2实战：手把手带你实现使用CenterNet2训练自定义数据集(1)
最新发布
m0_60634964的博客
05-15 346
Python崛起并且风靡，因为优点多、应用领域广、被大牛们认可。学习 Python 门槛很低，但它的晋级路线很多，通过它你能进入机器学习、数据挖掘、大数据，CS等更加高级的领域。Python可以做网络应用，可以做科学计算，数据分析，可以做网络爬虫，可以做机器学习、自然语言处理、可以写游戏、可以做桌面应用…Python可以做的很多，你需要学好基础，再选择明确的方向。这里给大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！
配置和运行CenterNet时踩过的坑
编程小白的博客
03-09 3787
在运行CenterNet时遇到的一些问题
CenterNet 编译DCNv2踩坑记录
weixin_44044604的博客
09-14 791
一直卡在DCNv2的编译上，尝试根据报错解决和查阅网上的各种教程均告失败，最后看到有人说DCNv2的版本要和torch版本对应 GitHub - lbin/DCNv2 at pytorch_1.6 把网址后面的版本改成自己的版本，下载后更换掉原先的DCNv2文件夹 就可以正常编译了 .........
DCNv2编译找不到nvcc问题
dragonfly91的博客
11-21 1532
在编译DCNv2 0.1版本的时候，遇到nvcc库找不到的问题，因为系统默认的nvcc库路径是/usr/bin/nvcc，此时需要手动设置相关的LD_LIBRARY_PATH。我的.bashrc的修改如下： export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_...
TensorRT-CenterNet.zip
12-14
1. **预训练模型**：CenterNet的预训练模型文件，通常是`.pth`或`.ckpt`格式，这些模型已经在大规模数据集如COCO或VOC上进行了训练，包含了对物体检测任务的学习权重。 2. **模型转换脚本**：TensorRT通常需要将...
CenterNet-master.zip
03-26
《CenterNet：目标检测新视角》 在计算机视觉领域，目标检测是一项至关重要的任务，它旨在识别并定位图像中的特定对象。近年来，随着深度学习技术的发展，目标检测算法取得了显著的进步。CenterNet便是其中的一种...
centernet源代码
11-13
2. **损失函数**：CenterNet使用联合损失函数来同时训练分类、中心点定位和大小估计。这通常包括点预测损失（如L1或Gaussian L2损失）和置信度损失（如交叉熵损失）。 3. **数据预处理**：数据预处理包括图像缩放、...
CenterNet目标检测网络几种针对COCO训练好的模型，Model
11-07
**中心点检测网络（CenterNet）与COCO数据集** **一、CenterNet目标检测网络** CenterNet是一种基于关键点的目标检测框架，由Ke Yang等人在2019年提出。它以其新颖的“中心点检测”理念而闻名，不同于传统的边界框...
错误make: nvcc: Command not found 解决方案
jndingxin的专栏
01-19 3534
系统：ubuntu18.04 错误现象： 在编译cu文件时，有时会遇到nvcc: Command not found的错误，系统已经安装了cuda，为什么会出现这个错误呢，大概率是环境变量没有配置，系统找不到nvcc命令所在路径。 解决方案： 打开/home/用户名/.bashrc文件 在文件的末尾添加如下内容： export PATH=/usr/local/cuda-10.2/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD.
DCNv2编译报错解决方法
dhdhhy的博客
08-21 582
【代码】DCNv2编译报错解决方法。
【实验记录】使用 CenterNet（Objects as Points）训练数据集
学习 & 分享 ~
09-01 1726
论文：https://arxiv.org/abs/1904.07850 代码：https://github.com/xingyizhou/CenterNet官方配置文档：CenterNet/readme/INSTALL.md官方在 这里 给的环境配置是 PyTorch 0.4.1 + CUDA 9.0 + CUDNN 7.1，但是我按照这个配出来的环境在跑训练时报了这个错： （不得不说这个错误非常气人！！）最开始找到的原因说 cuda 和 pytorch 版本有问题，要升级到 cuda 10.0 以上，但是
详解Centernet训练自己的数据集出现的问题（liunx+ cuda10 + pytorch1.0.1）
qq_33843251的博客
03-22 408
ps:最近在参加华为软挑，导师还布置了一大堆任务，题也没刷几个，抽时间贴一下自己走过的错路。 训练的流程就不讲了，见下面链接： https://blog.csdn.net/weixin_41765699/article/details/100118353 主要看里面出现错误，我们怎么解决的 1.## 错误： cannot import name 'imread' from 'scipy.misc' 这个出现在用vocl转coco数据集的过程中，主要是因为代码太老了，scipy里移除了imread; 解决办法
ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.
wzk4869的博客
12-31 2512
ImportError: torch.utils.ffi is deprecated. Please use cpp extensions instead.
（最新版本）如何在CenterNet上训练自己的数据集？
热门推荐
weixin_43509263的博客
09-13 2万+
一、git clone source code 从官方github仓库上把源码下下来，在pycharm上新建一个名称为“CenterNet”的工程，建好后的目录如下： 二、准备数据集 2-1 在data文件夹下新建两个文件夹：food、image_and_xml； 其中food为你自己数据集的名称，我这里要做的是识别一个菜品的任务，所以命名为food，image_and_xml存放的是你所有的图...
centernet训练内存拉满
06-08
Centernet 是一种目标检测算法，需要进行训练才能够达到较好的检测性能。如果在训练过程中出现内存占用过高的情况，可能是由于以下原因： 1. Batch size 过大：训练时每次输入的数据量过大，导致内存占用过高。可以...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值