https://github.com/machinelearningprodigy/objectdetection
This AI-powered app combines Stable Diffusion for image generation and DETR for object detection, all within an interactive Streamlit interface. 🎨🔍 Users can generate stunning AI images from text prompts and detect objects in uploaded images with high accuracy. 🚀 Simple, fast, and efficient—bringing AI creativity to your fingertips!
https://github.com/machinelearningprodigy/objectdetection
ai genai huggingface image-processing jupyter-notebook models python3
Last synced: 11 months ago
JSON representation
This AI-powered app combines Stable Diffusion for image generation and DETR for object detection, all within an interactive Streamlit interface. 🎨🔍 Users can generate stunning AI images from text prompts and detect objects in uploaded images with high accuracy. 🚀 Simple, fast, and efficient—bringing AI creativity to your fingertips!
- Host: GitHub
- URL: https://github.com/machinelearningprodigy/objectdetection
- Owner: machinelearningprodigy
- Created: 2024-08-24T10:17:04.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-02-13T16:33:41.000Z (12 months ago)
- Last Synced: 2025-02-13T17:34:13.274Z (12 months ago)
- Topics: ai, genai, huggingface, image-processing, jupyter-notebook, models, python3
- Language: Jupyter Notebook
- Homepage:
- Size: 81.1 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 🚀 AI Image Generation & Object Detection App
Welcome to the **AI Image Generation & Object Detection App**! This project utilizes **Stable Diffusion** for text-to-image generation and **DETR (DEtection TRansformer)** for object detection in images. 🎨🔍
---
## ✨ Features
✅ **Generate Images** from text prompts using Stable Diffusion 🖼️
✅ **Detect Objects** in uploaded images with DETR 🤖
✅ **Download Generated Images** with a single click ⬇️
✅ **Streamlit Web Interface** for an easy-to-use experience 🏆
---
## 📦 Installation
1. Clone this repository:
```bash
https://github.com/machinelearningprodigy/objectDetection
```
```bash
cd objectDetection
```
2. Install dependencies:
```bash
pip install -r requirements.txt
```
3. Download the Stable Diffusion model weights:
```bash
mkdir stable_diffusion_weights
```
Place your model weights inside the `stable_diffusion_weights` folder.
4. Run the app:
```bash
streamlit run app.py
```
---
## 🎨 Image Generation
1️⃣ Enter a **text prompt** (e.g., *"A futuristic city at sunset"*)
2️⃣ Click **Generate Image**
3️⃣ View and **download** your AI-generated masterpiece! 🌟
---
## 🔍 Object Detection
1️⃣ Upload an **image**
2️⃣ The model will **detect objects** and display bounding boxes 🖼️
3️⃣ View detected objects with labels and confidence scores 📊
---
## 🛠️ Technologies Used
- 🤖 **Stable Diffusion** (RunwayML Stable Diffusion v1-5)
- 🔍 **DETR (DEtection TRansformer)**
- 🚀 **Streamlit** (Interactive UI)
- 🏞️ **PIL (Pillow)** (Image Processing)
- 🔥 **PyTorch** (Deep Learning Framework)
---
## 🌟 Demo
Watch the AI generate images in real-time! 🎨
---
## 📜 License
This project is licensed under the **MIT License**. Feel free to use and modify it! 💡
💙 **Enjoy AI Creativity!** 🚀