Commit 3baefd0d authored by anjith2006's avatar anjith2006

Removing pretrained flag in deeppixbis extractor

parent 8cf3fe4a
Pipeline #36960 failed with stage
in 1 minute and 12 seconds
......@@ -41,7 +41,7 @@ class DeepPixBiSExtractor(Extractor):
# model
self.transforms = transforms
self.network = DeepPixBiS(pretrained=True)
self.network = DeepPixBiS(pretrained=False)
self.scoring_method = scoring_method
self.available_scoring_methods=['pixel_mean','binary','combined']
......
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