Create core/__init__.py
Browse files- core/__init__.py +83 -0
core/__init__.py
ADDED
|
@@ -0,0 +1,83 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Core processing modules for BackgroundFX Pro.
|
| 3 |
+
Handles main processing pipeline components.
|
| 4 |
+
"""
|
| 5 |
+
|
| 6 |
+
from .temporal import (
|
| 7 |
+
TemporalStabilizer,
|
| 8 |
+
TemporalConfig,
|
| 9 |
+
FrameBuffer,
|
| 10 |
+
FrameAnomalyDetector,
|
| 11 |
+
OpticalFlowTracker
|
| 12 |
+
)
|
| 13 |
+
|
| 14 |
+
from .models import (
|
| 15 |
+
ModelManager,
|
| 16 |
+
SAM2Model,
|
| 17 |
+
MatAnyoneModel,
|
| 18 |
+
ModelConfig,
|
| 19 |
+
ModelCache
|
| 20 |
+
)
|
| 21 |
+
|
| 22 |
+
from .quality import (
|
| 23 |
+
QualityAnalyzer,
|
| 24 |
+
QualityMetrics,
|
| 25 |
+
QualityConfig,
|
| 26 |
+
MetricsTracker,
|
| 27 |
+
QualityReport
|
| 28 |
+
)
|
| 29 |
+
|
| 30 |
+
from .edge import (
|
| 31 |
+
EdgeProcessor,
|
| 32 |
+
EdgeConfig,
|
| 33 |
+
HairSegmentation,
|
| 34 |
+
EdgeRefinement,
|
| 35 |
+
SymmetryCorrector
|
| 36 |
+
)
|
| 37 |
+
|
| 38 |
+
from .hair_segmentation import (
|
| 39 |
+
HairSegmentationPipeline,
|
| 40 |
+
HairConfig,
|
| 41 |
+
HairMaskRefiner,
|
| 42 |
+
AsymmetryDetector,
|
| 43 |
+
HairEdgeEnhancer
|
| 44 |
+
)
|
| 45 |
+
|
| 46 |
+
__version__ = "1.0.0"
|
| 47 |
+
|
| 48 |
+
__all__ = [
|
| 49 |
+
# Temporal
|
| 50 |
+
'TemporalStabilizer',
|
| 51 |
+
'TemporalConfig',
|
| 52 |
+
'FrameBuffer',
|
| 53 |
+
'FrameAnomalyDetector',
|
| 54 |
+
'OpticalFlowTracker',
|
| 55 |
+
|
| 56 |
+
# Models
|
| 57 |
+
'ModelManager',
|
| 58 |
+
'SAM2Model',
|
| 59 |
+
'MatAnyoneModel',
|
| 60 |
+
'ModelConfig',
|
| 61 |
+
'ModelCache',
|
| 62 |
+
|
| 63 |
+
# Quality
|
| 64 |
+
'QualityAnalyzer',
|
| 65 |
+
'QualityMetrics',
|
| 66 |
+
'QualityConfig',
|
| 67 |
+
'MetricsTracker',
|
| 68 |
+
'QualityReport',
|
| 69 |
+
|
| 70 |
+
# Edge
|
| 71 |
+
'EdgeProcessor',
|
| 72 |
+
'EdgeConfig',
|
| 73 |
+
'HairSegmentation',
|
| 74 |
+
'EdgeRefinement',
|
| 75 |
+
'SymmetryCorrector',
|
| 76 |
+
|
| 77 |
+
# Hair Segmentation
|
| 78 |
+
'HairSegmentationPipeline',
|
| 79 |
+
'HairConfig',
|
| 80 |
+
'HairMaskRefiner',
|
| 81 |
+
'AsymmetryDetector',
|
| 82 |
+
'HairEdgeEnhancer'
|
| 83 |
+
]
|