Unverified Commit 7bc5bd85 authored by Prabhat Roy's avatar Prabhat Roy Committed by GitHub
Browse files

Added download test for KITTI dataset (#3736)

parent 5bc68cc9
...@@ -392,6 +392,19 @@ def widerface(): ...@@ -392,6 +392,19 @@ def widerface():
) )
def kitti():
return itertools.chain(
*[
collect_download_configs(
lambda train=train: datasets.Kitti(ROOT, train=train, download=True),
name=f"Kitti, {'train' if train else 'test'}",
file="kitti",
)
for train in (True, False)
]
)
def make_parametrize_kwargs(download_configs): def make_parametrize_kwargs(download_configs):
argvalues = [] argvalues = []
ids = [] ids = []
...@@ -427,6 +440,7 @@ def make_parametrize_kwargs(download_configs): ...@@ -427,6 +440,7 @@ def make_parametrize_kwargs(download_configs):
usps(), usps(),
celeba(), celeba(),
widerface(), widerface(),
kitti(),
) )
) )
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment