From 2fb686a66a304f0a719bc53d77412460af97fd61 Mon Sep 17 00:00:00 2001 From: "Shariq F. Bhat" Date: Sat, 26 Dec 2020 19:11:25 +0300 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 091412d3..aa2b8fab 100644 --- a/README.md +++ b/README.md @@ -29,7 +29,7 @@ bin_edges, predicted_depth = model(example_rgb_batch) # KITTI model = UnetAdaptiveBins.build(n_bins=N_BINS, min_val=MIN_DEPTH, max_val=MAX_DEPTH_KITTI) -pretrained_path = "./pretrained/AdaBins_nyu.pt" +pretrained_path = "./pretrained/AdaBins_kitti.pt" model, _, _ = model_io.load_checkpoint(pretrained_path, model) bin_edges, predicted_depth = model(example_rgb_batch)