[Python] module ‘torchvision.models.detection’ has no attribute ‘retinanet_resnet50_fpn’

Python

問題内容

Pytorch内の学習済みモデルのRetinaNetをインポートしようと思ったら、RetinaNetが見つかりませんでした。普通にPytorchやtorchvisionを更新してもうまくいかなかったのですが、以下の通りにしたら、解決しました。

module 'torchvision.models.detection' has no attribute 'retinanet_resnet50_fpn'

解決策

以下のコマンドで、Pytorchとtorchvisionを同時に更新したら、うまく入りました。

pip install torch==1.7.1+cpu torchvision==0.8.2+cpu -f https://download.pytorch.org/whl/torch_stable.html
タイトルとURLをコピーしました