sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
209 stars 68 forks source link

ModuleNotFoundError: No module named 'vision_transformers' #39

Open samahwaleed opened 1 year ago

samahwaleed commented 1 year ago

I got this error and I tried to solve it by install vision-transformer, but still show me same error pip install vision-transformer-pytorch

sovit-123 commented 1 year ago

The Vision Transformer models are a work in progress and they depend on this library of mine https://github.com/sovit-123/vision_transformers It is recommended that you don't use the models now. I will update the README once I debug the detection pipeline for these models.

Sowgandh6 commented 1 year ago

Hey, Is the issue solved?

sovit-123 commented 1 year ago

@Sowgandh6 Yes, it is solved. You do not need to install vision_transformers to use Faster RCNN VitDet models. However, you will need to do pip install vision_transformers if you wish to use fasterrcnn_mobilevit_xxs.