chenfei-wu / TaskMatrix

Other
34.51k stars 3.32k forks source link

No module named 'groundingdino' #356

Closed fabien219200 closed 1 year ago

fabien219200 commented 1 year ago

After all the steps provided in the readme, I have this error when attempting to use the recommended cpu version image

Does anyone have the same problem ?

I also got this problem when I tried running it from Google Colab

Wang-Xiaodong1899 commented 1 year ago

Hi @fabien219200, we import the GroundingDINO and SAM modules by default, so, you should install these modules by this command: pip install git+https://github.com/IDEA-Research/GroundingDINO.git pip install git+https://github.com/facebookresearch/segment-anything.git After that, you can run" python visual_chatgpt.py --load "Text2Box_cuda:0,Segmenting_cuda:0, Inpainting_cuda:0,ImageCaptioning_cuda:0 Then, enjoy using the detection or segmentation-anything only by uploading an image and sending a sentence like "Find the cat in this image" or "segment the cat in this image". Thanks for your comment, we are updating the Colab code.

Wang-Xiaodong1899 commented 1 year ago

Hi @fabien219200 , The Colab notebook has been updated. Enjoy it!

fabien219200 commented 1 year ago

Great, works like a charm Thank you :)