From 0c1b3f9a72a72d42f5532c330c8318b9bbf60799 Mon Sep 17 00:00:00 2001
From: ageorge <anjith.george@idiap.ch>
Date: Wed, 22 May 2019 16:02:26 +0200
Subject: [PATCH] Added public link to docuumentation and small fix
---
README.rst | 6 +++---
bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py | 4 +++-
2 files changed, 6 insertions(+), 4 deletions(-)
diff --git a/README.rst b/README.rst
index e8d81af..097f436 100644
--- a/README.rst
+++ b/README.rst
@@ -2,9 +2,9 @@
.. Mon 03 Jul 2017 11:50:06 CEST
.. image:: http://img.shields.io/badge/docs-stable-yellow.svg
- :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
+ :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/stable/index.html
.. image:: http://img.shields.io/badge/docs-latest-orange.svg
- :target: http://beatubulatest.lab.idiap.ch/private/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
+ :target: http://idiap.ch/software/docs/bob/bob.paper.mccnn.tifs2018/master/index.html
.. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/build.svg
:target: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/commits/master
.. image:: https://gitlab.idiap.ch/bob/bob.paper.mccnn.tifs2018/badges/master/coverage.svg
@@ -80,4 +80,4 @@ development `mailing list`_.
.. _mailing list: https://www.idiap.ch/software/bob/discuss
.. _bob package development: https://www.idiap.ch/software/bob/docs/bob/bob.extension/master/
.. _conda: https://conda.io
-.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
\ No newline at end of file
+.. _install conda: https://conda.io/docs/install/quick.html#linux-miniconda-install
diff --git a/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py b/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
index 1ac899f..015c322 100644
--- a/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
+++ b/bob/paper/mccnn/tifs2018/trainer_configs/wmca_mccnn.py
@@ -70,7 +70,9 @@ SELECTED_CHANNELS = [0, 1, 2, 3]
img_transform_train = transforms.Compose([ChannelSelect(selected_channels=SELECTED_CHANNELS), RandomHorizontalFlipImage(p=0.5), transforms.ToTensor()])
-dataset = DataFolder(data_folder=data_folder_train,
+dataset={}
+
+dataset['train'] = DataFolder(data_folder=data_folder_train,
transform=img_transform_train,
extension='.hdf5',
bob_hldi_instance=bob_hldi_instance_train,
--
GitLab