优化

5cdbaddd · zhengyaoqiu · 72b87b4e · 5cdbaddd
Commit 5cdbaddd authored Jun 04, 2025 by zhengyaoqiu
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

feature_extractor.py app/services/feature_extractor.py +1 -1

No files found.
--- a/app/services/feature_extractor.py
+++ b/app/services/feature_extractor.py
@@ -20,7 +20,7 @@ class FeatureExtractor:
        device = "cpu"
        self.model, self.preprocess = self.init_model(device, model_name)
        self.device = device
-        model_path = Path(__file__).parent.absolute() / "../models/checkpoints/exp-schp-201908261155-lip.pth"
+        model_path = Path(__file__).parent.absolute() / "../models/exp-schp-201908261155-lip.pth"
        self.schp_model = HumanParsingModel(model_path, dataset='lip')

    @staticmethod