diff --git a/README.md b/README.md index 2005ede..2be51a2 100644 --- a/README.md +++ b/README.md @@ -57,14 +57,12 @@ This python project is organized as follows: ## Usage Examples Here we report the example tests on the 3D hyperbolic data included in the paper. - ```bash # solve mask 1 saving the CNN weights python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot1.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --savemodel --outdir TL/shot1 # solve mask 2 from scratch python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot2.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --outdir TL/shot2_scratch # solve mask 2 using as initial guess the CNN weights of mask 1 python main.py --imgdir ./datasets/hyperbolic3d --imgname original.npy --maskname random66_shot2.npy --datadim 3d --gain 40 --upsample linear --epochs 3000 --outdir TL/shot2_transfer --net load --netdir TL/shot1/model.pth - ``` #### Data preparation We are glad you want to try our method on your data! To minimize the effort, keep in mind that: