X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=README.md;h=85e7df60d2d8cd78f2b793c26fc9163c810ed457;hp=f4e07af22d33e897e1f457d43d81a78dfd3387c3;hb=3c9b04dccaaf2a42cca35d5ea266f442cbb726ea;hpb=de4d7faef08d682d83c075253e532af54fd39c45 diff --git a/README.md b/README.md index f4e07af..85e7df6 100644 --- a/README.md +++ b/README.md @@ -13,7 +13,9 @@ make -j -k ./test-svrt.py ``` -should generate an image example.png in the current directory. +should generate an image +[`example.png`](https://fleuret.org/git-extract/pysvrt/example.png) in +the current directory. Note that the image generation does not take advantage of GPUs or multi-core, and can be as fast as 10,000 vignettes per second and as @@ -23,21 +25,22 @@ slow as 40 on a 4GHz i7-6700K. ## Vignette sets ## -The svrtset.py implements the classes `VignetteSet` and -`CompressedVignetteSet` with the following constructor +The file [`svrtset.py`](https://fleuret.org/git-extract/pysvrt/svrtset.py) implements the classes `VignetteSet` and +`CompressedVignetteSet` both with a constructor ``` __init__(problem_number, nb_samples, batch_size, cuda = False, logger = None) ``` -and the following method to return one batch +and a method ``` (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). +which returns 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 ## @@ -83,7 +86,8 @@ 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 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. +The file +[`cnn-svrt.py`](https://fleuret.org/git-extract/pysvrt/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.