🚀 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:

   https://github.com/machinelearningprodigy/objectDetection
   

   cd objectDetection

2. Install dependencies:

   pip install -r requirements.txt

3. Download the Stable Diffusion model weights:

   mkdir stable_diffusion_weights

   Place your model weights inside the stable_diffusion_weights folder.

4. Run the app:

   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! 🚀