OpenGVLab / InternGPT

пятница, 26 мая 2023 г. в 00:00:11

InternGPT (iGPT) is an open source demo platform where you can easily showcase your AI models. Now it supports DragGAN, ChatGPT, ImageBind, multimodal chat like GPT-4, SAM, interactive image editing, etc. Try it at igpt.opengvlab.com

[中文文档]

The project is still under construction, we will continue to update it and welcome contributions/pull requests from the community.

🤖💬 InternGPT [Paper]

InternGPT(short for iGPT) / InternChat(short for iChat) is pointing-language-driven visual interactive system, allowing you to interact with ChatGPT by clicking, dragging and drawing using a pointing device. The name InternGPT stands for interaction, nonverbal, and ChatGPT. Different from existing interactive systems that rely on pure language, by incorporating pointing instructions, iGPT significantly improves the efficiency of communication between users and chatbots, as well as the accuracy of chatbots in vision-centric tasks, especially in complicated visual scenarios. Additionally, in iGPT, an auxiliary control mechanism is used to improve the control capability of LLM, and a large vision-language model termed Husky is fine-tuned for high-quality multi-modal dialogue (impressing ChatGPT-3.5-turbo with 93.89% GPT-4 Quality).

🥳 🚀 What's New

(2023.05.24) 🎉🎉🎉 We have supported the DragGAN! Please see the video demo for the usage. Let's try this awesome feauture: Demo.
(2023.05.18) We have supported ImageBind. Please see the video demo for the usage.
(2023.05.15) The model_zoo including HuskyVQA has been released! Try it on your local machine!
(2023.05.15) Our code is also publicly available on Hugging Face! You can duplicate the repository and run it on your own GPUs.

🤖💬 Online Demo

InternGPT is online (see https://igpt.opengvlab.com). Let's try it!

[NOTE] It is possible that you are waiting in a lengthy queue. You can clone our repo and run it with your private GPU.

🧭 User Manual

Update:

(2023.05.24) We now support DragGAN. You can try it as follows:

Click the button New Image;
Click the image where blue denotes the start point and red denotes the end point;
Notice that the number of blue points is the same as the number of red points. Then you can click the button Drag It;
After processing, you will receive an edited image and a video that visualizes the editing process.

(2023.05.18) We now support ImageBind. If you want to generate a new image conditioned on audio, you can upload an audio file in advance:

To generate a new image from a single audio file, you can send the message like: "generate a real image from this audio";
To generate a new image from audio and text, you can send the message like: "generate a real image from this audio and {your prompt}";
To generate a new image from audio and image, you need to upload an image and then send the message like: "generate a new image from above image and audio".

Main features:

After uploading the image, you can have a multi-modal dialogue by sending messages like: "what is it in the image?" or "what is the background color of image?".
You also can interactively operate, edit or generate the image as follows:

You can click the image and press the button Pick to visualize the segmented region or press the button OCR to recognize the words at chosen position;
To remove the masked reigon in the image, you can send the message like: "remove the masked region";
To replace the masked reigon in the image, you can send the message like: "replace the masked region with {your prompt}";
To generate a new image, you can send the message like: "generate a new image based on its segmentation describing {your prompt}"
To create a new image by your scribble, you should press button Whiteboard and draw in the board. After drawing, you need to press the button Save and send the message like: "generate a new image based on this scribble describing {your prompt}".

Video Demo with DragGAN:

dragGAN_demo2.mp4

Video Demo with ImageBind:

video_demo_with_imagebind.mp4

iGPT Video Demo:

online_demo.mp4

🗓️ Schedule

🏠 System Overview

🎁 Major Features

Remove the masked object

Interactive image editing

Image generation

Interactive visual question answer

Interactive image generation

Video highlight interpretation

🛠️ Installation

See INSTALL.md

👨‍🏫 Get Started

Running the following shell can start a gradio service:

python -u app.py --load "HuskyVQA_cuda:0,SegmentAnything_cuda:0,ImageOCRRecognition_cuda:0" --port 3456

if you want to enable the voice assistant, please use openssl to generate the certificate:

mkdir certificate
openssl req -x509 -newkey rsa:4096 -keyout certificate/key.pem -out certificate/cert.pem -sha256 -days 365 -nodes

and then run:

python -u app.py --load "HuskyVQA_cuda:0,SegmentAnything_cuda:0,ImageOCRRecognition_cuda:0" --port 3456 --https

🎫 License

This project is released under the Apache 2.0 license.

🖊️ Citation

If you find this project useful in your research, please consider cite:

@article{2023interngpt,
  title={InternGPT: Solving Vision-Centric Tasks by Interacting with ChatGPT Beyond Language},
  author={Liu, Zhaoyang and He, Yinan and Wang, Wenhai and Wang, Weiyun and Wang, Yi and Chen, Shoufa and Zhang, Qinglong and Yang, Yang and Li, Qingyun and Yu, Jiashuo and others},
  journal={arXiv preprint arXiv:2305.05662},
  year={2023}
}

🤝 Acknowledgement

Thanks to the open source of the following projects:

Hugging Face LangChain TaskMatrix SAM Stable Diffusion ControlNet InstructPix2Pix BLIP Latent Diffusion Models EasyOCR ImageBind DragGAN

Welcome to discuss with us and continuously improve the user experience of InternGPT.

WeChat QR Code