X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=README.md;h=f4e07af22d33e897e1f457d43d81a78dfd3387c3;hp=d0ec215a1767bd07bfc53b0e0758ce1891c39bd1;hb=de4d7faef08d682d83c075253e532af54fd39c45;hpb=77153ad6f6acb94a5132e9930722500cd93a6960

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.