ModuleNotFoundError: No module named 'detectron2.modeling.backbone.vit'

[HenryGou]

When I was fine-tuning on COCO, I got an error as shown in the title. Although I followed the instructions, Detectron2 may not have been installed successfully. Could you suggest something?

[wangsr126]

If you are able to successfully import detectron2 but facing issues with importing detectron2.modeling.backbone.vit, you may consider building Detectron2 from source as some of the latest features in the main branch may not be included in the pre-built packages. However, if you are unable to import detectron2, it would be advisable to refer to the Detectron2 repository for assistance.

[HenryGou]

Thank you so much for your response, which has helped me successfully resolve the issue. Could you please provide guidance on how to train a segmentation task using my own dataset? I would greatly appreciate your assistance.

If you are able to successfully import detectron2 but facing issues with importing detectron2.modeling.backbone.vit, you may consider building Detectron2 from source as some of the latest features in the main branch may not be included in the pre-built packages. However, if you are unable to import detectron2, it would be advisable to refer to the Detectron2 repository for assistance.

[wangsr126]

We are also working on it and will release the related code and results once ready.

[HenryGou]

Thanks. Wishing you great success.