This commit is contained in:
KaiyangZhou 2018-05-25 16:52:26 +01:00
parent e50179f4bc
commit d45d779cc1

View File

@ -1841,7 +1841,4 @@ def init_img_dataset(name, **kwargs):
def init_vid_dataset(name, **kwargs):
if name not in __vid_factory.keys():
raise KeyError("Invalid dataset, got '{}', but expected to be one of {}".format(name, __vid_factory.keys()))
return __vid_factory[name](**kwargs)
if __name__ == '__main__':
dataset = iLIDS()
return __vid_factory[name](**kwargs)