minor change

This commit is contained in:
Kai Zhang 2020-11-01 18:00:56 -05:00 committed by GitHub
parent a6800157c7
commit 42353ef775
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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: