DETR 教程可用于自定义数据 :)

几个月前,我已经将 DETR 添加到 HuggingFace Transformers 中,我已经替换了原始的 torchvision 后端(ResNets),这样您就可以使用 timm 仓库中提供的 任何 后端(如 EfficientNets 或 MobileNets 等)。该模型使用与 HuggingFace 中其他模型(如 BERT)相同的 API 实现(即您可以使用 DetrModel(这是一个没有任何头部的编码器-解码器 Transformer)、DetrForObjectDetection(其中包含对象检测和类标签分类器的头部)以及 DetrForSegmentation(其中添加了掩码头部)。