Spaces:
Sleeping
Sleeping
Update landmarkdiff/synthetic/__init__.py to v0.3.2
Browse files
landmarkdiff/synthetic/__init__.py
CHANGED
|
@@ -1,23 +1,7 @@
|
|
| 1 |
-
"""Synthetic data generation for
|
| 2 |
-
|
| 3 |
-
Modules:
|
| 4 |
-
- pair_generator: Generate training pairs from face images
|
| 5 |
-
- augmentation: Clinical degradation augmentations
|
| 6 |
-
- tps_warp: TPS warping with rigid region preservation
|
| 7 |
-
"""
|
| 8 |
-
|
| 9 |
-
from landmarkdiff.synthetic.augmentation import apply_clinical_augmentation
|
| 10 |
-
from landmarkdiff.synthetic.pair_generator import (
|
| 11 |
-
TrainingPair,
|
| 12 |
-
generate_pair,
|
| 13 |
-
generate_pairs_from_directory,
|
| 14 |
-
)
|
| 15 |
-
from landmarkdiff.synthetic.tps_warp import warp_image_tps
|
| 16 |
|
| 17 |
__all__ = [
|
| 18 |
-
"
|
| 19 |
-
"
|
| 20 |
-
"
|
| 21 |
-
"generate_pairs_from_directory",
|
| 22 |
-
"warp_image_tps",
|
| 23 |
]
|
|
|
|
| 1 |
+
"""Synthetic training data generation for LandmarkDiff."""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
__all__ = [
|
| 4 |
+
"augmentation",
|
| 5 |
+
"pair_generator",
|
| 6 |
+
"tps_warp",
|
|
|
|
|
|
|
| 7 |
]
|