Update README.md

shariqfarooq123 · Dec 26, 2020 · 2fb686a · 2fb686a
1 parent 301c6c1
commit 2fb686a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)