112 lines
6.1 KiB
Python
112 lines
6.1 KiB
Python
from tests.image_generation.helpers.image_generation_fibo_test_helper import ImageGeneratorFiboTestHelper
|
|
|
|
OWL_PROMPT = """
|
|
{
|
|
"short_description": "A hyper-detailed, ultra-fluffy owl sitting in the trees at night, looking directly at the camera with wide, adorable, expressive eyes. Its feathers are soft and voluminous, catching the cool moonlight with subtle silver highlights. The owl's gaze is curious and full of charm, giving it a whimsical, storybook-like personality.",
|
|
"objects": [
|
|
{
|
|
"description": "An adorable, fluffy owl with large, expressive eyes and soft, voluminous feathers. Its plumage is a mix of warm browns, grays, and subtle silver highlights from the moonlight.",
|
|
"location": "center",
|
|
"relationship": "The owl is the sole subject, perched comfortably within its environment.",
|
|
"relative_size": "large within frame",
|
|
"shape_and_color": "Round head, large eyes, bulky body, predominantly brown and grey with silver accents.",
|
|
"texture": "Extremely soft, fluffy, and detailed feathers, giving a plush toy-like appearance.",
|
|
"appearance_details": "The eyes are wide, dark, and reflective, conveying a sense of wonder and curiosity. The beak is small and light-colored, almost hidden by the feathers. Subtle silver highlights catch the moonlight on its feathers.",
|
|
"orientation": "upright, facing forward"
|
|
}
|
|
],
|
|
"background_setting": "A dark, nocturnal forest setting with blurred trees and foliage, illuminated by a soft, cool moonlight. The background is out of focus, emphasizing the owl.",
|
|
"lighting": {
|
|
"conditions": "moonlight",
|
|
"direction": "backlit and side-lit from the left",
|
|
"shadows": "soft, diffused shadows on the right side of the owl and within the background foliage, indicating a single light source."
|
|
},
|
|
"aesthetics": {
|
|
"composition": "centered, portrait composition",
|
|
"color_scheme": "cool blues and silvers from the moonlight contrasting with warm browns and grays of the owl and forest.",
|
|
"mood_atmosphere": "mysterious, enchanting, whimsical, and serene.",
|
|
"aesthetic_score": "very high",
|
|
"preference_score": "very high"
|
|
},
|
|
"photographic_characteristics": {
|
|
"depth_of_field": "shallow",
|
|
"focus": "sharp focus on the owl's face and eyes, with a soft blur in the background.",
|
|
"camera_angle": "eye-level",
|
|
"lens_focal_length": "portrait lens (e.g., 50mm-85mm)"
|
|
},
|
|
"style_medium": "digital illustration",
|
|
"text_render": [],
|
|
"context": "A whimsical character illustration, possibly for a children's book, animated film, or fantasy art collection.",
|
|
"artistic_style": "fantasy, illustrative, detailed"
|
|
}
|
|
"""
|
|
|
|
OWL_PROMPT_REFINED = """
|
|
{
|
|
"short_description": "A hyper-detailed, ultra-fluffy owl sitting in the trees at night, looking directly at the camera with wide, adorable, expressive eyes. Its feathers are soft and voluminous, catching the cool moonlight with subtle silver highlights. The owl's gaze is curious and full of charm, giving it a whimsical, storybook-like personality.",
|
|
"objects": [
|
|
{
|
|
"description": "An adorable, fluffy owl with large, expressive eyes and soft, voluminous feathers. Its plumage is a mix of white and subtle silver highlights from the moonlight.",
|
|
"location": "center",
|
|
"relationship": "The owl is the sole subject, perched comfortably within its environment.",
|
|
"relative_size": "large within frame",
|
|
"shape_and_color": "Round head, large eyes, bulky body, predominantly white with silver accents.",
|
|
"texture": "Extremely soft, fluffy, and detailed feathers, giving a plush toy-like appearance.",
|
|
"appearance_details": "The eyes are wide, dark, and reflective, conveying a sense of wonder and curiosity. The beak is small and light-colored, almost hidden by the feathers. Subtle silver highlights catch the moonlight on its feathers.",
|
|
"orientation": "upright, facing forward"
|
|
}
|
|
],
|
|
"background_setting": "A dark, nocturnal forest setting with blurred trees and foliage, illuminated by a soft, cool moonlight. The background is out of focus, emphasizing the owl.",
|
|
"lighting": {
|
|
"conditions": "moonlight",
|
|
"direction": "backlit and side-lit from the left",
|
|
"shadows": "soft, diffused shadows on the right side of the owl and within the background foliage, indicating a single light source."
|
|
},
|
|
"aesthetics": {
|
|
"composition": "centered, portrait composition",
|
|
"color_scheme": "cool blues and silvers from the moonlight contrasting with white of the owl and forest.",
|
|
"mood_atmosphere": "mysterious, enchanting, whimsical, and serene.",
|
|
"aesthetic_score": "very high",
|
|
"preference_score": "very high"
|
|
},
|
|
"photographic_characteristics": {
|
|
"depth_of_field": "shallow",
|
|
"focus": "sharp focus on the owl's face and eyes, with a soft blur in the background.",
|
|
"camera_angle": "eye-level",
|
|
"lens_focal_length": "portrait lens (e.g., 50mm-85mm)"
|
|
},
|
|
"style_medium": "digital illustration",
|
|
"text_render": [],
|
|
"context": "A whimsical character illustration, possibly for a children's book, animated film, or fantasy art collection.",
|
|
"artistic_style": "fantasy, illustrative, detailed"
|
|
}
|
|
"""
|
|
|
|
|
|
class TestImageGeneratorFibo:
|
|
def test_image_generation_fibo(self):
|
|
ImageGeneratorFiboTestHelper.assert_matches_reference_image(
|
|
reference_image_path="reference_fibo.png",
|
|
output_image_path="output_fibo.png",
|
|
prompt=OWL_PROMPT, # Assume this has been generated by the VLM, actual VLM tests are separate
|
|
steps=20,
|
|
seed=42,
|
|
height=176,
|
|
width=320,
|
|
guidance=4.0,
|
|
quantize=8,
|
|
)
|
|
|
|
def test_image_generation_fibo_refined_white_owl(self):
|
|
ImageGeneratorFiboTestHelper.assert_matches_reference_image(
|
|
reference_image_path="reference_fibo_white_owl.png",
|
|
output_image_path="output_fibo_white_owl.png",
|
|
prompt=OWL_PROMPT_REFINED, # Assume this has been refined by the VLM, actual VLM tests are separate
|
|
steps=20,
|
|
seed=42,
|
|
height=176,
|
|
width=320,
|
|
guidance=4.0,
|
|
quantize=8,
|
|
)
|