From 42353ef775f0bf27c9b84941e80934009d4f9ad9 Mon Sep 17 00:00:00 2001
From: Kai Zhang <kz298@cornell.edu>
Date: Sun, 1 Nov 2020 18:00:56 -0500
Subject: [PATCH] minor change

---
 data_loader_split.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/data_loader_split.py b/data_loader_split.py
index 1fd49b6..8fb7bd8 100644
--- a/data_loader_split.py
+++ b/data_loader_split.py
@@ -31,6 +31,9 @@ def load_data_split(basedir, scene, split, skip=1, try_load_min_depth=True, only
         nums = open(filename).read().split()
         return np.array([float(x) for x in nums]).reshape([4, 4]).astype(np.float32)
 
+    if basedir[-1] == '/':          # remove trailing '/'
+        basedir = basedir[:-1]
+     
     split_dir = '{}/{}/{}'.format(basedir, scene, split)
 
     if only_img_files: