Pretrained model path templates

bob/paper/mccnn/tifs2018/config/FASNet_config.py

@@ -96,8 +96,35 @@ from bob.learn.pytorch.extractor.image import FASNetExtractor
 
 from bob.bio.video.extractor import Wrapper
 
-MODEL_FILE= None # Replace with '<PATH_TO_MODEL>'   
-####################################################################
+# MODEL_FILE= None # Replace with '<PATH_TO_MODEL>'   
+# ####################################################################
+
+# If you want to use the pretrained model
+
+import pkg_resources
+
+MODEL_FILE = pkg_resources.resource_filename( 'bob.paper.mccnn.tifs2018', 'models/fasnet.pth')
+
+URL='http://www.idiap.ch/~ageorge/model_100_0.pth'
+
+if not os.path.exists(MODEL_FILE):
+
+  logger.info('Downloading the FASNet model')
+
+  bob.extension.download.download_file(URL,MODEL_FILE)
+
+  logger.info('Downloaded FASNet model to location: {}'.format(MODEL_FILE))
+
+
+
+
+
+MODEL_FILE=
+
+/idiap/temp/ageorge/MCCNN_paperpackage/bob.paper.mccnn.tifs2018/src/bob.paper.mccnn.tifs2018/bob/paper/mccnn/tifs2018/models/mccnn_best_C1-B1-FFC.pth
+/idiap/temp/ageorge/MCCNN_paperpackage/bob.paper.mccnn.tifs2018/src/bob.paper.mccnn.tifs2018/bob/paper/mccnn/tifs2018/models/fasnet.pth
+
+
 
 
 _img_transform = transforms.Compose([transforms.ToPILImage(),transforms.Resize(224, interpolation=2),transforms.ToTensor(),transforms.Normalize(mean=[0.485, 0.456, 0.406],

bob/paper/mccnn/tifs2018/config/MCCNN_config.py

@@ -92,10 +92,27 @@ from bob.learn.pytorch.extractor.image import MCCNNExtractor
 
 from bob.bio.video.extractor import Wrapper
 
-MODEL_FILE= None # Replace with '<PATH_TO_MODEL>'   
-####################################################################
+# MODEL_FILE= None # Replace with '<PATH_TO_MODEL>'   
+# ####################################################################
+
+# If you want to use the pretrained model
+
+import pkg_resources
+
+MODEL_FILE = pkg_resources.resource_filename( 'bob.paper.mccnn.tifs2018', 'models/mccnn_best_C1-B1-FFC.pth')
+
+URL='http://www.idiap.ch/~ageorge/model_100_0.pth'
+
+if not os.path.exists(MODEL_FILE):
+
+  logger.info('Downloading the MCCNN model')
+
+  bob.extension.download.download_file(URL,MODEL_FILE)
+
+  logger.info('Downloaded MCCNN model to location: {}'.format(MODEL_FILE))
+
 
-ADAPTED_LAYERS= 'conv1-group1-block1-ffc'
+ADAPTED_LAYERS= 'conv1-group1-ffc'
 ####################################################################
 
 SELECTED_CHANNELS= [0,1,2,3]