Deformable-DETR:可变形的DETR

时间:2024-03-30 15:56:25
【文件属性】:

文件名称:Deformable-DETR:可变形的DETR

文件大小:388KB

文件格式:ZIP

更新时间:2024-03-30 15:56:25

Python

可变形的DETR 由,, ,,,。 该存储库是论文《可的正式实现。 介绍 TL; 博士可变形DETR是一种高效且快速收敛的端到端对象检测器。 它通过一种新颖的基于采样的有效注意力机制,缓解了DETR的高复杂性和缓慢收敛的问题。 抽象的。 最近提出了DETR,以消除物体检测中对许多手工设计组件的需求,同时表现出良好的性能。 但是,由于Transformer注意模块在处理图像特征图时的局限性,它收敛缓慢且特征空间分辨率有限。 为了缓解这些问题,我们提出了可变形DETR,其关注模块仅关注参考周围的一小部分关键采样点。 可变形的DETR可以比DETR(尤其是在小物体上)获得更好的性能,训练时间减少10倍。 在COCO基准上进行的大量实验证明了我们方法的有效性。 执照 该项目是根据。 变更日志 有关主要更改的详细日志,请参见 。 引用可变形DETR 如果您发现可变形DETR在您的研究中很有用


【文件预览】:
Deformable-DETR-main
----benchmark.py(2KB)
----requirements.txt(30B)
----datasets()
--------transforms.py(9KB)
--------samplers.py(5KB)
--------__init__.py(1KB)
--------data_prefetcher.py(3KB)
--------coco_panoptic.py(4KB)
--------torchvision_datasets()
--------coco.py(6KB)
--------panoptic_eval.py(2KB)
--------coco_eval.py(9KB)
----figs()
--------convergence.png(53KB)
--------illustration.png(266KB)
----models()
--------__init__.py(591B)
--------deformable_transformer.py(17KB)
--------matcher.py(5KB)
--------ops()
--------backbone.py(5KB)
--------deformable_detr.py(23KB)
--------position_encoding.py(4KB)
--------segmentation.py(16KB)
----LICENSE(12KB)
----engine.py(7KB)
----util()
--------__init__.py(506B)
--------box_ops.py(3KB)
--------misc.py(17KB)
--------plot_utils.py(5KB)
----tools()
--------launch.py(9KB)
--------run_dist_launch.sh(812B)
--------run_dist_slurm.sh(1KB)
----README.md(10KB)
----configs()
--------r50_deformable_detr_plus_iterative_bbox_refinement.sh(198B)
--------r50_deformable_detr_single_scale_dc5.sh(206B)
--------r50_deformable_detr.sh(143B)
--------r50_deformable_detr_single_scale.sh(185B)
--------r50_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh(236B)
----main.py(15KB)
----docs()
--------changelog.md(383B)

网友评论