a3924ae4a8
- Added intelligent quality checkers and auto-retry pipeline for `image-to-3d` and `text-to-3d`. - Added unit tests for quality checkers. - `text-to-3d` now supports more `text-to-image` models, pipeline success rate improved to 94%.
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
# Project EmbodiedGen
|
|
#
|
|
# Copyright (c) 2025 Horizon Robotics. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied. See the License for the specific language governing
|
|
# permissions and limitations under the License.
|
|
|
|
import logging
|
|
|
|
from embodied_gen.validators.aesthetic_predictor import AestheticPredictor
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# @pytest.mark.manual
|
|
def test_aesthetic_predictor():
|
|
image_path = "apps/assets/example_image/sample_02.jpg"
|
|
predictor = AestheticPredictor(device="cpu")
|
|
score = predictor.predict(image_path)
|
|
|
|
assert isinstance(score, float)
|
|
logger.info(f"Aesthetic score: {score:.3f}")
|