README.md

metadata

title: FastAPI Model
sdk: docker
emoji: 🚀
colorFrom: blue
colorTo: purple
pinned: false

<<<<<<< HEAD

AI-Powered Body Measurement & Apparel Sizing

This project is a FastAPI-based AI system that predicts body measurements and recommends clothing sizes using deep learning. It processes front and side images, removes backgrounds, and utilizes a trained Keras model for accurate predictions.

---

📌 Features

✅ AI-powered body measurement estimation using images.
✅ Automatic background removal for clean image processing.
✅ Deep learning-based predictions using a trained model.
✅ FastAPI integration for quick and efficient API responses.

---

🚀 Installation

To set up the project, follow these steps:

1️⃣ Clone the Repository

git clone https://github.com/CloozyBrands/AI-BodyMeasurement.git

---

2️⃣ Install Dependencies

pip install -r requirements.txt

3️⃣ Run the FastAPI Server

uvicorn app.main:app --reload

Then open http://127.0.0.1:8000/docs to test the API.

---

🛠️ How It Works

1️⃣ Upload Images

• The API accepts two images:
  • Front view
  • Side view

2️⃣ AI-Based Processing

• The images are processed using single_person_processor.py, which:
  ✅ Removes the background using rembg.
  ✅ Converts the images into a model-compatible format.
  ✅ Predicts body measurements like chest, waist, and height.

3️⃣ Clothing Size Prediction

• Based on the body measurements, the system suggests a T-shirt and pants size using predefined size charts.

---

🔍 API Endpoints

1️⃣ /predict/ (POST) - Predict Body Measurements

📌 Example Request:

POST /predict/

📌 Request Parameters:

Parameter	Type	Description
front_image	File	Front view image (JPEG/PNG)
side_image	File	Side view image (JPEG/PNG)
input_data	JSON	User data (height, weight, gender)

📌 Example JSON Payload:

{
  "gender": 0, 
  "height_cm": 175, 
  "weight_kg": 70, 
  "apparel_type": "all"
}

📌 Example Response:

{
  "results": {
    "body_measurements": {
      "chest": 100.5,
      "waist": 80.2,
      "hip": 97.3
    },
    "tshirt_size": "L",
    "pants_size": 34
  }
}

---

🎯 Model & AI Processing

📌 The model used in this project is a TensorFlow/Keras model stored as best_model.keras.
📌 The AI processing is handled inside single_person_processor.py, which:

• Loads the trained model using tf.keras.models.load_model.
• Extracts measurements based on input images.
• Maps the measurements to standard clothing sizes.

---

📜 License

This project is licensed under the MIT License – feel free to modify and use it.

---

🤝 Contributing

1. Fork the repository
2. Create a new branch
3. Make your changes
4. Submit a pull request

We welcome contributions and improvements! 🚀

>>>>>>> 4e80050 (FastAPI model for size recommendation)