Generowanie obrazów za pomocą Imagen

Imagen to model generowania obrazów. Może służyć do generowania niestandardowych awatarów w profilach użytkowników lub do integrowania spersonalizowanych komponentów wizualnych z istniejącymi ścieżkami ekranów, aby zwiększać zaangażowanie użytkowników.

Dostęp do modeli Imagen możesz uzyskać z aplikacji na Androida za pomocą pakietu SDK Firebase AI Logic. Modele Imagen są dostępne za pomocą dostawców interfejsów API Firebase AI Logic: Gemini Developer API (zalecany dla większości deweloperów) i Vertex AI.

Diagram ilustrujący architekturę integracji Firebase AI Logic w celu uzyskania dostępu do interfejsu Gemini Developer API. Aplikacja na Androida korzysta z pakietu Firebase Android SDK, aby łączyć się z Firebase. Firebase wchodzi wtedy w interakcję z interfejsem Gemini Developer API, który uzyskuje dostęp do Gemini Pro i Flash w chmurze.
Rysunek 1. Dostęp do modeli Imagen za pomocą Firebase AI Logic.

Eksperymentuj z promptami

Stworzenie idealnych promptów często wymaga wielu prób. Możesz eksperymentować z promptami dotyczącymi obrazów w Vertex AI Studio, środowisku IDE do projektowania i prototypowania promptów. Wskazówki dotyczące ulepszania promptów znajdziesz w przewodniku po promptach i atrybutach obrazów.

Zrzut ekranu interfejsu Google Cloud Vertex AI Studio przedstawiający 4 wygenerowane obrazy tyranozaura z niebieskim plecakiem w prehistorycznym lesie. Na pasku bocznym po lewej stronie znajdują się opcje nawigacji, takie jak Panel, Model Garden i Vision, a w panelu po prawej stronie są widoczne parametry ustawień generowania obrazów.
Rysunek 2. Vertex AI Studio może pomóc Ci dopracować prompty generowania obrazów.

Konfigurowanie projektu Firebase i łączenie z nim aplikacji

Wykonaj czynności opisane w dokumentacji Firebase, aby dodać Firebase do projektu na Androida.

Dodaj zależność Gradle

Dodaj do pliku build.gradle te zależności:

dependencies {
  // Import the BoM for the Firebase platform
  implementation(platform("com.google.firebase:firebase-bom:34.2.0"))

  // Add the dependency for the Firebase AI Logic library. When using the BoM,
  // you don't specify versions in Firebase library dependencies
  implementation("com.google.firebase:firebase-ai")
}

Generowanie obrazu

Aby wygenerować obraz w aplikacji na Androida, zacznij od utworzenia instancji klasy ImagenModel z opcjonalną konfiguracją.

Za pomocą parametru generationConfig możesz zdefiniować prompt ujemny, liczbę obrazów, format obrazu wyjściowego, format obrazu i dodać znak wodny. Za pomocą parametru safetySettings możesz skonfigurować filtry bezpieczeństwa i osób.

Kotlin

val config = ImagenGenerationConfig {
    numberOfImages = 2,
    aspectRatio = ImagenAspectRatio.LANDSCAPE_16x9,
    imageFormat = ImagenImageFormat.jpeg(compressionQuality = 100),
    addWatermark = false
}

// Initialize the Gemini Developer API backend service
// For Vertex AI use Firebase.ai(backend = GenerativeBackend.vertexAI())
val model = Firebase.ai(backend = GenerativeBackend.googleAI()).imagenModel(
    modelName = "imagen-3.0-generate-002",
    generationConfig = config,
    safetySettings = ImagenSafetySettings(
       safetyFilterLevel = ImagenSafetyFilterLevel.BLOCK_LOW_AND_ABOVE,
       personFilterLevel = ImagenPersonFilterLevel.BLOCK_ALL
    )
)

Java

ImagenGenerationConfig config = new ImagenGenerationConfig.Builder()
    .setNumberOfImages(2)
    .setAspectRatio(ImagenAspectRatio.LANDSCAPE_16x9)
    .setImageFormat(ImagenImageFormat.jpeg(100))
    .setAddWatermark(false)
    .build();

// For Vertex AI use Firebase.ai(backend = GenerativeBackend.vertexAI())
ImagenModelFutures model = ImagenModelFutures.from(
    FirebaseAI.ai(backend = GenerativeBackend.googleAI()).imagenModel(
       "imagen-3.0-generate-002",
       config,
       ImagenSafetySettings.builder()
          .setSafetyFilterLevel(ImagenSafetyFilterLevel.BLOCK_LOW_AND_ABOVE)
          .setPersonFilterLevel(ImagenPersonFilterLevel.BLOCK_ALL)
          .build())
);

Po utworzeniu instancji ImagenModel możesz generować obrazy, wywołując:generateImages

Kotlin

val imageResponse = model.generateImages(
  prompt = "An astronaut riding a horse",
)
val image = imageResponse.images.first
val bitmapImage = image.asBitmap()

Java

CompletableFuture<GenerateContentResponse> futureResponse =
    model.generateContent(
        Content.newBuilder()
            .addParts(
                Part.newBuilder()
                    .setText("An astronaut riding a horse")
                    .build())
            .build());

try {
  GenerateContentResponse imageResponse = futureResponse.get();
  List<GeneratedImage> images =
      imageResponse
          .getCandidates(0)
          .getContent()
          .getParts(0)
          .getInlineData()
          .getImagesList();

  if (!images.isEmpty()) {
    GeneratedImage image = images.get(0);
    Bitmap bitmapImage = image.asBitmap();
    // Use bitmapImage
  }
} catch (ExecutionException | InterruptedException e) {
  e.printStackTrace();
}