Commit ffa579c1 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

Merge branch 'densenet' into 'master'

Stop Downloading Densenet in nosetests

See merge request !41
parents 360ebedb 386fabde
Pipeline #37185 passed with stages
in 10 minutes and 26 seconds
...@@ -41,7 +41,7 @@ class DeepPixBiSExtractor(Extractor): ...@@ -41,7 +41,7 @@ class DeepPixBiSExtractor(Extractor):
# model # model
self.transforms = transforms self.transforms = transforms
self.network = DeepPixBiS(pretrained=True) self.network = DeepPixBiS(pretrained=False)
self.scoring_method = scoring_method self.scoring_method = scoring_method
self.available_scoring_methods=['pixel_mean','binary','combined'] 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