先前的DN-DETR模型都是在服务器上运行的,后来在本地运行时出现了一些小问题,这篇博文则主要介绍DN-DETR模型在本地运行时所需要做的配置。
运行环境
首先DN-DETR的运行环境与DINO一致,这里就不再赘述了。
博主使用的本地配置是I7-13700H+4060(8G)
调试过程
在github上下载源码后,在idea中打开,事实上我们应该安装readme的过程来进行调试,但由于博主先前已经都调试过了,这里就默认环境已经都配置好了。
我们直接运行main.py即可,这里需要设定几个参数:
首先是确定我们要运行哪个模型,DN-DETR
是在DETR
基础上设计的,其融合了前人的研究成果,在DAB-DETR
,Deformable-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,所以不要以为是错误,让子弹飞一会。