From de4d7faef08d682d83c075253e532af54fd39c45 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Mon, 19 Jun 2017 14:41:57 +0200 Subject: [PATCH] Update. --- README.md | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/README.md b/README.md index d0ec215..f4e07af 100644 --- a/README.md +++ b/README.md @@ -36,6 +36,9 @@ and the following method to return one batch (torch.FloatTensor, torch.LongTensor) get_batch(b) ``` +as a pair composed of a 4d 'input' Tensor (i.e. single channel 128x128 +images), and a 1d 'target' Tensor (i.e. Boolean labels). + ## Low-level functions ## The main function for genering vignettes is @@ -80,15 +83,7 @@ See vignette_set.py for a class CompressedVignetteSet using it. # Testing convolution networks # -The file - -``` -cnn-svrt.py -``` - -provides the implementation of two deep networks, and use the -compressed vignette code to allow the training with several millions -vignettes on a PC with 16Gb and a GPU with 8Gb. - -The networks were designed by Afroze Baqapuri during an internship at -Idiap. +The file `cnn-svrt.py` provides the implementation of two deep +networks designed by Afroze Baqapuri during an internship at Idiap, +and allows to train them with several millions vignettes on a PC with +16Gb and a GPU with 8Gb. -- 2.20.1