From 8f27e779d51a804948eac64d0719ffd233c30bbf Mon Sep 17 00:00:00 2001
From: ageorge <anjith.george@idiap.ch>
Date: Fri, 3 May 2019 14:53:41 +0200
Subject: [PATCH] Pretrained model path templates
---
.../mccnn/tifs2018/config/FASNet_config.py | 31 +++++++++++++++++--
.../mccnn/tifs2018/config/MCCNN_config.py | 23 ++++++++++++--
bob/paper/mccnn/tifs2018/models/__init__.py | 0
3 files changed, 49 insertions(+), 5 deletions(-)
create mode 100644 bob/paper/mccnn/tifs2018/models/__init__.py
diff --git a/bob/paper/mccnn/tifs2018/config/FASNet_config.py b/bob/paper/mccnn/tifs2018/config/FASNet_config.py
index b0f805b..2acbff1 100644
--- a/bob/paper/mccnn/tifs2018/config/FASNet_config.py
+++ b/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],
diff --git a/bob/paper/mccnn/tifs2018/config/MCCNN_config.py b/bob/paper/mccnn/tifs2018/config/MCCNN_config.py
index 71e02fc..d698bcc 100644
--- a/bob/paper/mccnn/tifs2018/config/MCCNN_config.py
+++ b/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]
diff --git a/bob/paper/mccnn/tifs2018/models/__init__.py b/bob/paper/mccnn/tifs2018/models/__init__.py
new file mode 100644
index 0000000..e69de29
--
GitLab