AttributeError：torchvision.models.detection无deformable_detr属性求助

可能的原因及修复步骤 #

1. 检查torchvision版本兼容性 #

DeformableDETR是torchvision较新版本才引入的模型，需确保你的torchvision版本≥0.14.0（对应PyTorch≥1.13.0）。可通过以下代码查看当前版本：

print(torchvision.__version__)

若版本过低，执行升级命令：

pip install --upgrade torch torchvision

2. 修正模型创建方式 #

不要直接实例化DeformableDETR类，torchvision提供了封装好的工厂函数来创建模型，正确写法为：

detr = torchvision.models.detection.deformable_detr(num_classes=1)

结合你已有的导入语句，也可简化为：

detr = detection.deformable_detr(num_classes=1)

3. （可选）直接调用类的正确方式 #

若你确实需要直接实例化类（不推荐，工厂函数会自动处理默认参数与权重加载逻辑），需确认类的正确导入路径，修改导入语句后再创建模型：

from torchvision.models.detection.deformable_detr import DeformableDETR
detr = DeformableDETR(num_classes=1)

注意：这种方式需要手动配置所有必要参数（如backbone、num_queries等），操作成本更高。

内容的提问来源于stack exchange，提问作者Huy1LapTrinh