Skip to content

I
image_search
Commit 5cdbaddd authored by zhengyaoqiu's avatar zhengyaoqiu
Browse files
Options

优化

parent 72b87b4e
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
app/services/feature_extractor.py
View file @ 5cdbaddd
...@@ -20,7 +20,7 @@ class FeatureExtractor: ...@@ -20,7 +20,7 @@ class FeatureExtractor:
device = "cpu" device = "cpu"
self.model, self.preprocess = self.init_model(device, model_name) self.model, self.preprocess = self.init_model(device, model_name)
self.device = device 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') self.schp_model = HumanParsingModel(model_path, dataset='lip')
@staticmethod @staticmethod
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment