Skip to content

Commit e736da0

Browse files
author
Clément Pinard
committed
prevent mpi both to be selected with random split
1 parent 8753096 commit e736da0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

datasets/mpisintel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,8 @@ def mpi_sintel_both(root, transform=None, target_transform=None,
5959
'''load images from both clean and final folders.
6060
We cannot shuffle input, because it would very likely cause data snooping
6161
for the clean and final frames are not that different'''
62+
assert(isinstance(split, str)), 'To avoid data snooping, you must provide a static list of train/val when dealing with both clean and final.'
63+
' Look at Sintel_train_val.txt for an example'
6264
train_list1, test_list1 = make_dataset(root, split, 'clean')
6365
train_list2, test_list2 = make_dataset(root, split, 'final')
6466
train_dataset = ListDataset(root, train_list1 + train_list2, transform, target_transform, co_transform)

0 commit comments

Comments
 (0)