Skip to content

Commit 3bbfe4e

Browse files
authored
Update dataloader.py
ignore not included view
1 parent fb30a93 commit 3bbfe4e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

dataset/dataloader.py

+2
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,8 @@ def __init__(self,params):
4646
continue
4747
root_path_ = os.path.join(root_path,drone)
4848
for view in os.listdir(root_path_):
49+
if view not in self.view_list:
50+
continue
4951
view_path = os.path.join(root_path_,view)
5052
#get included types
5153
for type_ in os.listdir(view_path):

0 commit comments

Comments
 (0)