From 6671c470a48d72ea255cb98633547fc792f742b1 Mon Sep 17 00:00:00 2001 From: TimNiklasWitte Date: Thu, 31 Mar 2022 14:29:26 +0200 Subject: [PATCH] expand readme --- README.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index afa1153..feb5afa 100644 --- a/README.md +++ b/README.md @@ -92,6 +92,9 @@ python3 live_recolor_plot.py ### Pretrained Model The model was runned for 13 epochs on the `imagnet2012` dataset and its weights are stored in `./saved_models`. + +![Train and test loss](./Plots/TrainTestLossPlot.png) + Note that, grey images must have a shape of `(256,256,1)`. The following code will load the pretrained model and colorized an image: @@ -105,5 +108,5 @@ grey_img = ... # grey_img.shape = (256,256,1) grey_img = np.expand_dims(grey_img, axis=0) # add batch dim ab_img = autoencoder(grey_img) # get ab values -rbg_img = getRGB(grey_img, ab_img) # contained in Main.py +rbg_img = getRGB(grey_img, ab_img) # contained in Training.py ```