Clearer example usage of saved checkpoint

Jeffrey-Ede · Oct 20, 2020 · 16f3224 · 16f3224
1 parent 80c3e90
commit 16f3224
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -34,8 +34,11 @@ from denoiser import Denoiser, disp
 #Try replacing this with your own image!
 img = np.random.rand(1500, 1500)
 
+#Replace with the location of your saved checkpoint
+checkpoint_loc = "//flexo.ads.warwick.ac.uk/Shared41/Microscopy/Jeffrey-Ede/models/denoiser-multi-gpu-13/model"
+
 #Initialize the denoising neural network
-noise_remover = Denoiser()
+noise_remover = Denoiser(checkpoint_loc=checkpoint_loc)
 
 #Denoise a 512x512 crop from the image
 crop = img[:512,:512]