Tanishq71's picture
Update README.md
8762064 verified
|
Raw
History Blame Contribute Delete
2.13 kB
metadata
title: OpenEnv Image Optimizer
emoji: 🚀
colorFrom: blue
colorTo: green
sdk: docker
app_port: 7860
tags:
  - openenv

OpenEnv: Automated Image Augmentation Optimizer

Description and Motivation

In modern MLOps pipelines, applying static image augmentations can destroy critical features. This OpenEnv simulates an MLOps pipeline where an agent dynamically sequences corrective operations to maximize downstream classifier accuracy, framing data-cleaning as a Reinforcement Learning problem.

Observation and Action Spaces

Observation Space: A strict JSON state containing image metrics: avg_brightness, noise_variance, contrast_ratio, and simulated current_accuracy. Action Space: A JSON command to apply operations: increase_brightness, decrease_brightness, apply_denoise, increase_contrast, or submit_pipeline, alongside an intensity float [0.1 - 1.0].

Tasks & Expected Difficulty

  1. task_1_easy_brightness (Easy): Image is heavily underexposed. The agent must increase brightness without blowing out highlights.
  2. task_2_medium_noise (Medium): Severe static noise. The agent must balance aggressive denoising with contrast enhancement.
  3. task_3_hard_pipeline (Hard): Multi-variable corruption (underexposed, noisy, washed out). Requires strict sequencing of multiple operations.

Setup & Environment Variables

The inference script strictly requires the OpenAI Python Client and reads the following environment variables:

  • HF_TOKEN: Mandatory. Your Hugging Face or OpenAI API key.
  • API_BASE_URL: The API endpoint for the LLM (Defaults to https://api.openai.com/v1).
  • MODEL_NAME: The model identifier (Defaults to gpt-4o-mini).

Run via Docker: docker run -e HF_TOKEN="your_key" <image_name>

Run Locally: python inference.py (Ensure HF_TOKEN is exported in your terminal).

Baseline Scores

The baseline execution utilizes gpt-4o-mini and outputs the strictly required [START], [STEP], and [END] telemetry.

  • task_1_easy_brightness: 0.92 / 1.0
  • task_2_medium_noise: 0.85 / 1.0
  • task_3_hard_pipeline: 0.81 / 1.0