DN-DETR调试记录

先前的DN-DETR模型都是在服务器上运行的,后来在本地运行时出现了一些小问题,这篇博文则主要介绍DN-DETR模型在本地运行时所需要做的配置。

运行环境

首先DN-DETR的运行环境与DINO一致,这里就不再赘述了。
博主使用的本地配置是I7-13700H+4060(8G)

调试过程

在github上下载源码后,在idea中打开,事实上我们应该安装readme的过程来进行调试,但由于博主先前已经都调试过了,这里就默认环境已经都配置好了。
我们直接运行main.py即可,这里需要设定几个参数:
首先是确定我们要运行哪个模型,DN-DETR是在DETR基础上设计的,其融合了前人的研究成果,在DAB-DETRDeformable-DETR等模型基础上进行改进,这里这个modelname则是告诉我们选择哪个。

parser.add_argument('--modelname', '-m', type=str, default="dn_dab_detr", choices=['dn_dab_detr', 'dn_dab_deformable_detr',                                                              'dn_dab_deformable_detr_deformable_encoder_only', 'dn_dab_dino_deformable_detr'])

随后是数据集地址,这里博主使用的是先前自己构造的KITTI数据集

 parser.add_argument('--coco_path', default="D:\graduate\datasets\detection\kitti/",type=str, )

随后是num_classes参数的设定,在对应的model的文件中,如这里博主选择的是DN-DAB-DETR,那么我们就去对应的模型文件中找即可。
这里全部改为对应的类别数+1即可,博主的数据集中有3个类别,再加一个背景类则为4个,如下图所示:

在这里插入图片描述
此时可以运行试一下,发现报页面空间不足的问题,这时只需要修改num_workers值即可,博主改为4。

随后运行再次报错:

targets, scalar, label_noise_scale, box_noise_scale, num_patterns = dn_args
TypeError: cannot unpack non-iterable NoneType object

这个报错似乎是迭代对象为空,事实上是因为main.py文件中的use_dn参数的设定导致的,我们将其设置为True即可

 parser.add_argument('--use_dn', default="True",
                        help="use denoising training.")

再次运行就OK了。
博主的batch-size为2,此时模型的显存占用情况如下:

在这里插入图片描述

此外,DN-DAB-DETR在开始训练时的值是很低的,接近0,所以不要以为是错误,让子弹飞一会。

相关推荐

  1. 查找DNS解析记录

    2023-12-25 13:12:02       16 阅读
  2. DN-DETR的原理和源码解读

    2023-12-25 13:12:02       10 阅读
  3. LVGL调试记录

    2023-12-25 13:12:02       6 阅读
  4. 5. DNS 记录和报文

    2023-12-25 13:12:02       9 阅读
  5. DNS记录类型介绍(A记录、MX记录、NS记录等)

    2023-12-25 13:12:02       30 阅读
  6. 【<span style='color:red;'>DNS</span>】

    DNS

    2023-12-25 13:12:02      33 阅读

最近更新

  1. TCP协议是安全的吗?

    2023-12-25 13:12:02       17 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2023-12-25 13:12:02       16 阅读
  3. 【Python教程】压缩PDF文件大小

    2023-12-25 13:12:02       15 阅读
  4. 通过文章id递归查询所有评论(xml)

    2023-12-25 13:12:02       18 阅读

热门阅读

  1. 常用git指令

    2023-12-25 13:12:02       34 阅读
  2. 2023年底和2024年节假日及补班日期

    2023-12-25 13:12:02       93 阅读
  3. 门面设计模式

    2023-12-25 13:12:02       36 阅读
  4. Linux系统中跟TCP相关的内核参数

    2023-12-25 13:12:02       35 阅读
  5. vscode windows下 tasks.json 和 launch.json

    2023-12-25 13:12:02       36 阅读
  6. SQL分类

    SQL分类

    2023-12-25 13:12:02      30 阅读
  7. mysql全局事务变量GTID

    2023-12-25 13:12:02       23 阅读
  8. leetcode 131. 分割回文串

    2023-12-25 13:12:02       36 阅读
  9. [网络安全] NTFS权限

    2023-12-25 13:12:02       42 阅读
  10. 《PCI Express体系结构导读》随记 —— 前言

    2023-12-25 13:12:02       32 阅读
  11. Mybatis使用详解

    2023-12-25 13:12:02       40 阅读
  12. Linux 文件管理命令----pwd 命令

    2023-12-25 13:12:02       38 阅读