Trang này mô tả cách sử dụng tính năng inpainting bằng Imagen để chèn một đối tượng vào hình ảnh bằng cách sử dụng SDK Firebase AI Logic.
Inpainting là một loại chỉnh sửa dựa trên mặt nạ. Mặt nạ là một lớp phủ kỹ thuật số xác định vùng cụ thể mà bạn muốn chỉnh sửa.
Cách hoạt động: Bạn cung cấp một hình ảnh gốc và một hình ảnh tương ứng đã được che phủ bằng mặt nạ (do hệ thống tự động tạo hoặc do bạn cung cấp) để xác định mặt nạ trên một vùng mà bạn muốn thêm nội dung mới. Bạn cũng cung cấp một câu lệnh dạng văn bản mô tả nội dung bạn muốn thêm. Sau đó, mô hình sẽ tạo và thêm nội dung mới trong vùng đã che phủ bằng mặt nạ.
Ví dụ: bạn có thể che phủ một chiếc bàn bằng mặt nạ và yêu cầu mô hình thêm một bình hoa.
Chuyển đến mã cho mặt nạ được tạo tự động Chuyển đến mã để cung cấp mặt nạ
Trước khi bắt đầu
| Chỉ dùng được khi bạn sử dụng Vertex AI Gemini API làm nhà cung cấp API. |
Nếu chưa hoàn tất, hãy làm theo
hướng dẫn bắt đầu. Hướng dẫn này
mô tả cách thiết lập dự án Firebase, kết nối ứng dụng với Firebase,
thêm SDK, khởi chạy dịch vụ phụ trợ cho nhà cung cấp API mà bạn chọn và
tạo một thực thể ImagenModel.
Các mô hình hỗ trợ tính năng này
Imagen cung cấp tính năng chỉnh sửa hình ảnh thông qua mô hình capability:
imagen-3.0-capability-001
Xin lưu ý rằng đối với các mô hình Imagen, vị trí global
không được hỗ trợ.
Chèn đối tượng bằng mặt nạ được tạo tự động
| Trước khi thử mẫu này, hãy hoàn tất phần Trước khi bắt đầu của hướng dẫn này để thiết lập dự án và ứng dụng. |
Mẫu sau đây cho biết cách sử dụng tính năng inpainting để chèn nội dung vào hình ảnh bằng cách tự động tạo mặt nạ. Bạn cung cấp hình ảnh gốc và một câu lệnh dạng văn bản, sau đó Imagen tự động phát hiện và tạo một vùng mặt nạ để sửa đổi hình ảnh gốc.
Swift
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho Swift. Hãy quay lại kiểm tra sau trong năm nay!
Kotlin
Để chèn đối tượng bằng mặt nạ được tạo tự động, hãy chỉ định ImagenBackgroundMask. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.INPAINT_INSERTION.
// Using this SDK to access Imagen models is a Preview release and requires opt-in
@OptIn(PublicPreviewAPI::class)
suspend fun customizeImage() {
// Initialize the Vertex AI Gemini API backend service
// Optionally specify the location to access the model (for example, `us-central1`)
val ai = Firebase.ai(backend = GenerativeBackend.vertexAI(location = "us-central1"))
// Create an `ImagenModel` instance with an Imagen "capability" model
val model = ai.imagenModel("imagen-3.0-capability-001")
// This example assumes 'originalImage' is a pre-loaded Bitmap.
// In a real app, this might come from the user's device or a URL.
val originalImage: Bitmap = TODO("Load your original image Bitmap here")
// Provide the prompt describing the content to be inserted.
val prompt = "a vase of flowers on the table"
// Use the editImage API to insert the new content.
// Pass the original image, the prompt, and an editing configuration.
val editedImage = model.editImage(
referenceImages = listOf(
ImagenRawImage(originalImage.toImagenInlineImage()),
ImagenBackgroundMask(), // Use ImagenBackgroundMask() to auto-generate the mask.
),
prompt = prompt,
// Define the editing configuration for inpainting and insertion.
config = ImagenEditingConfig(ImagenEditMode.INPAINT_INSERTION)
)
// Process the resulting 'editedImage' Bitmap, for example, by displaying it in an ImageView.
}
Java
Để chèn đối tượng bằng mặt nạ được tạo tự động, hãy chỉ định ImagenBackgroundMask. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.INPAINT_INSERTION.
// Initialize the Vertex AI Gemini API backend service
// Optionally specify the location to access the model (for example, `us-central1`)
// Create an `ImagenModel` instance with an Imagen "capability" model
ImagenModel imagenModel = FirebaseAI.getInstance(GenerativeBackend.vertexAI("us-central1"))
.imagenModel(
/* modelName */ "imagen-3.0-capability-001");
ImagenModelFutures model = ImagenModelFutures.from(imagenModel);
// This example assumes 'originalImage' is a pre-loaded Bitmap.
// In a real app, this might come from the user's device or a URL.
Bitmap originalImage = null; // TODO("Load your image Bitmap here");
// Provide the prompt describing the content to be inserted.
String prompt = "a vase of flowers on the table";
// Define the list of sources for the editImage call.
// This includes the original image and the auto-generated mask.
ImagenRawImage rawOriginalImage =
new ImagenRawImage(ImagenInlineImageKt.toImagenInlineImage(originalImage));
// Use ImagenBackgroundMask() to auto-generate the mask.
ImagenBackgroundMask rawMaskedImage = new ImagenBackgroundMask();
ImagenEditingConfig config = new ImagenEditingConfig();
// Use the editImage API to insert the new content.
// Pass the original image, the auto-generated masked image, the prompt, and an editing configuration.
Futures.addCallback(model.editImage(Arrays.asList(rawOriginalImage, rawMaskedImage), prompt, config),
new FutureCallback<ImagenGenerationResponse>() {
@Override
public void onSuccess(ImagenGenerationResponse result) {
if (result.getImages().isEmpty()) {
Log.d("ImageEditor", "No images generated");
}
Bitmap editedImage = ((ImagenInlineImage) result.getImages().get(0)).asBitmap();
// Process and use the bitmap to display the image in your UI
}
@Override
public void onFailure(Throwable t) {
// ...
}
}, Executors.newSingleThreadExecutor());
Web
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho ứng dụng web. Hãy quay lại kiểm tra sau trong năm nay!
Dart
Để chèn đối tượng bằng mặt nạ được tạo tự động, hãy chỉ định ImagenBackgroundMask. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.inpaintInsertion.
import 'dart:typed_data';
import 'package:firebase_ai/firebase_ai.dart';
import 'package:firebase_core/firebase_core.dart';
import 'firebase_options.dart';
// Initialize FirebaseApp
await Firebase.initializeApp(
options: DefaultFirebaseOptions.currentPlatform,
);
// Initialize the Vertex AI Gemini API backend service
// Optionally specify a location to access the model (for example, `us-central1`)
final ai = FirebaseAI.vertexAI(location: 'us-central1');
// Create an `ImagenModel` instance with an Imagen "capability" model
final model = ai.imagenModel(model: 'imagen-3.0-capability-001');
// This example assumes 'originalImage' is a pre-loaded Uint8List.
// In a real app, this might come from the user's device or a URL.
final Uint8List originalImage = Uint8List(0); // TODO: Load your original image data here.
// Provide the prompt describing the content to be inserted.
final prompt = 'a vase of flowers on the table';
try {
// Use the editImage API to insert the new content.
// Pass the original image, the prompt, and an editing configuration.
final response = await model.editImage(
sources: [
ImagenRawImage(originalImage),
ImagenBackgroundMask(), // Use ImagenBackgroundMask() to auto-generate the mask.
],
prompt: prompt,
// Define the editing configuration for inpainting and insertion.
config: const ImagenEditingConfig(
editMode: ImagenEditMode.inpaintInsertion,
),
);
// Process the result.
if (response.images.isNotEmpty) {
final editedImage = response.images.first.bytes;
// Use the editedImage (a Uint8List) to display the image, save it, etc.
print('Image successfully generated!');
} else {
// Handle the case where no images were generated.
print('Error: No images were generated.');
}
} catch (e) {
// Handle any potential errors during the API call.
print('An error occurred: $e');
}
Unity
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho Unity. Hãy quay lại kiểm tra sau trong năm nay!
Chèn đối tượng bằng mặt nạ do bạn cung cấp
| Trước khi thử mẫu này, hãy hoàn tất phần Trước khi bắt đầu của hướng dẫn này để thiết lập dự án và ứng dụng. |
Mẫu sau đây cho biết cách sử dụng tính năng inpainting để chèn nội dung vào hình ảnh bằng mặt nạ được xác định trong hình ảnh mà bạn cung cấp. Bạn cung cấp hình ảnh gốc, một câu lệnh dạng văn bản và hình ảnh đã che phủ bằng mặt nạ.
Swift
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho Swift. Hãy quay lại kiểm tra sau trong năm nay!
Kotlin
Để chèn đối tượng và cung cấp hình ảnh đã che phủ bằng mặt nạ của riêng bạn, hãy chỉ định ImagenRawMask bằng hình ảnh đã che phủ bằng mặt nạ. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.INPAINT_INSERTION.
// Using this SDK to access Imagen models is a Preview release and requires opt-in
@OptIn(PublicPreviewAPI::class)
suspend fun customizeImage() {
// Initialize the Vertex AI Gemini API backend service
// Optionally specify the location to access the model (for example, `us-central1`)
val ai = Firebase.ai(backend = GenerativeBackend.vertexAI(location = "us-central1"))
// Create an `ImagenModel` instance with an Imagen "capability" model
val model = ai.imagenModel("imagen-3.0-capability-001")
// This example assumes 'originalImage' is a pre-loaded Bitmap.
// In a real app, this might come from the user's device or a URL.
val originalImage: Bitmap = TODO("Load your original image Bitmap here")
// This example assumes 'maskImage' is a pre-loaded Bitmap that contains the masked area.
// In a real app, this might come from the user's device or a URL.
val maskImage: Bitmap = TODO("Load your masked image Bitmap here")
// Provide the prompt describing the content to be inserted.
val prompt = "a vase of flowers on the table"
// Use the editImage API to insert the new content.
// Pass the original image, the masked image, the prompt, and an editing configuration.
val editedImage = model.editImage(
referenceImages = listOf(
ImagenRawImage(originalImage.toImagenInlineImage()),
ImagenRawMask(maskImage.toImagenInlineImage()), // Use ImagenRawMask() to provide your own masked image.
),
prompt = prompt,
// Define the editing configuration for inpainting and insertion.
config = ImagenEditingConfig(ImagenEditMode.INPAINT_INSERTION)
)
// Process the resulting 'editedImage' Bitmap, for example, by displaying it in an ImageView.
}
Java
Để chèn đối tượng và cung cấp hình ảnh đã che phủ bằng mặt nạ của riêng bạn, hãy chỉ định ImagenRawMask bằng hình ảnh đã che phủ bằng mặt nạ. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.INPAINT_INSERTION.
// Initialize the Vertex AI Gemini API backend service
// Optionally specify the location to access the model (for example, `us-central1`)
// Create an `ImagenModel` instance with an Imagen "capability" model
ImagenModel imagenModel = FirebaseAI.getInstance(GenerativeBackend.vertexAI("us-central1"))
.imagenModel(
/* modelName */ "imagen-3.0-capability-001");
ImagenModelFutures model = ImagenModelFutures.from(imagenModel);
// This example assumes 'originalImage' is a pre-loaded Bitmap.
// In a real app, this might come from the user's device or a URL.
Bitmap originalImage = null; // TODO("Load your original image Bitmap here");
// This example assumes 'maskImage' is a pre-loaded Bitmap that contains the masked area.
// In a real app, this might come from the user's device or a URL.
Bitmap maskImage = null; // TODO("Load your masked image Bitmap here");
// Provide the prompt describing the content to be inserted.
String prompt = "a vase of flowers on the table";
// Define the list of source images for the editImage call.
ImagenRawImage rawOriginalImage =
new ImagenRawImage(ImagenInlineImageKt.toImagenInlineImage(originalImage));
// Use ImagenRawMask() to provide your own masked image.
ImagenBackgroundMask rawMaskedImage =
new ImagenRawMask(ImagenInlineImageKt.toImagenInlineImage(maskImage));
ImagenEditingConfig config = new ImagenEditingConfig();
// Use the editImage API to insert the new content.
// Pass the original image, the masked image, the prompt, and an editing configuration.
Futures.addCallback(model.editImage(Arrays.asList(rawOriginalImage, rawMaskedImage), prompt, config), new FutureCallback<ImagenGenerationResponse>() {
@Override
public void onSuccess(ImagenGenerationResponse result) {
if (result.getImages().isEmpty()) {
Log.d("ImageEditor", "No images generated");
}
Bitmap editedImage = ((ImagenInlineImage) result.getImages().get(0)).asBitmap();
// Process and use the bitmap to display the image in your UI
}
@Override
public void onFailure(Throwable t) {
// ...
}
}, Executors.newSingleThreadExecutor());
Web
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho ứng dụng web. Hãy quay lại kiểm tra sau trong năm nay!
Dart
Để chèn đối tượng và cung cấp hình ảnh đã che phủ bằng mặt nạ của riêng bạn, hãy chỉ định ImagenRawMask bằng hình ảnh đã che phủ bằng mặt nạ. Sử dụng
editImage()
và đặt cấu hình chỉnh sửa để sử dụng ImagenEditMode.inpaintInsertion.
import 'dart:typed_data';
import 'package:firebase_ai/firebase_ai.dart';
import 'package:firebase_core/firebase_core.dart';
import 'firebase_options.dart';
// Initialize FirebaseApp
await Firebase.initializeApp(
options: DefaultFirebaseOptions.currentPlatform,
);
// Initialize the Vertex AI Gemini API backend service
// Optionally specify a location to access the model (for example, `us-central1`)
final ai = FirebaseAI.vertexAI(location: 'us-central1');
// Create an `ImagenModel` instance with an Imagen "capability" model
final model = ai.imagenModel(model: 'imagen-3.0-capability-001');
// This example assumes 'originalImage' is a pre-loaded Uint8List.
// In a real app, this might come from the user's device or a URL.
final Uint8List originalImage = Uint8List(0); // TODO: Load your original image data here.
// This example assumes 'maskImage' is a pre-loaded Uint8List that contains the masked area.
// In a real app, this might come from the user's device or a URL.
final Uint8List maskImage = Uint8List(0); // TODO: Load your masked image data here.
// Provide the prompt describing the content to be inserted.
final prompt = 'a vase of flowers on the table';
try {
// Use the editImage API to insert the new content.
// Pass the original image, the prompt, and an editing configuration.
final response = await model.editImage(
sources: [
ImagenRawImage(originalImage),
ImagenRawMask(maskImage), // Use ImagenRawMask() to provide your own masked image.
],
prompt: prompt,
// Define the editing configuration for inpainting and insertion.
config: const ImagenEditingConfig(
editMode: ImagenEditMode.inpaintInsertion,
),
);
// Process the result.
if (response.images.isNotEmpty) {
final editedImage = response.images.first.bytes;
// Use the editedImage (a Uint8List) to display the image, save it, etc.
print('Image successfully generated!');
} else {
// Handle the case where no images were generated.
print('Error: No images were generated.');
}
} catch (e) {
// Handle any potential errors during the API call.
print('An error occurred: $e');
}
Unity
Tính năng chỉnh sửa hình ảnh bằng các mô hình Imagen không được hỗ trợ cho Unity. Hãy quay lại kiểm tra sau trong năm nay!
Các phương pháp hay nhất và hạn chế
Bạn nên mở rộng mặt nạ khi chỉnh sửa hình ảnh. Việc này có thể giúp làm mượt
các đường viền của bản chỉnh sửa và khiến bản chỉnh sửa trông thuyết phục hơn. Thông thường, bạn nên sử dụng giá trị mở rộng
là 1% hoặc 2% (0.01 hoặc 0.02).
Gửi ý kiến phản hồi về trải nghiệm của bạn với Firebase AI Logic