41 lines
1008 B
Python
41 lines
1008 B
Python
"""
|
|
Utility functions for FASHN VTON.
|
|
|
|
This package provides common utilities:
|
|
- Common Python helpers (exists, default, cast_tuple, compact)
|
|
- Model checkpoint loading
|
|
- Tensor operations and conversions
|
|
- Sampling schedules for Rectified Flow
|
|
- Pose keypoint utilities
|
|
- Logging setup
|
|
"""
|
|
|
|
from .checkpoint import load_checkpoint
|
|
from .common import cast_tuple, compact, default, exists
|
|
from .keypoints import get_dummy_dw_keypoints
|
|
from .logger import setup_logger
|
|
from .sampling import get_rf_schedule, time_shift
|
|
from .tensor import normalize_uint8_to_neg1_1, numpy_to_torch, tensor_to_pil, unpack_images
|
|
|
|
__all__ = [
|
|
# Common helpers
|
|
"exists",
|
|
"default",
|
|
"cast_tuple",
|
|
"compact",
|
|
# Checkpoint loading
|
|
"load_checkpoint",
|
|
# Tensor operations
|
|
"numpy_to_torch",
|
|
"unpack_images",
|
|
"normalize_uint8_to_neg1_1",
|
|
"tensor_to_pil",
|
|
# Sampling
|
|
"time_shift",
|
|
"get_rf_schedule",
|
|
# Pose
|
|
"get_dummy_dw_keypoints",
|
|
# Logging
|
|
"setup_logger",
|
|
]
|