If a flist, contains just one file (.txt) path, the load_flist function in dataset.py returns a 0-dimensional numpy array. As a result, the landmarks = np.genfromtxt(self.landmark_data[index]) command (line 105) in dataset.py fails, because a 0-dimensional numpy array can't be indexed.
To fix this, the load_flist function must return a numpy array with at least one dimension.
Type of change
Bug fix
Description
If a flist, contains just one file (.txt) path, the load_flist function in dataset.py returns a 0-dimensional numpy array. As a result, the
landmarks = np.genfromtxt(self.landmark_data[index])command (line 105) in dataset.py fails, because a 0-dimensional numpy array can't be indexed.To fix this, the load_flist function must return a numpy array with at least one dimension.