widerface

pytorch · Feb 2, 2021 · c8f54c4 · c8f54c4
1 parent 44997b0
commit c8f54c4
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/test_datasets_download.py b/test/test_datasets_download.py
@@ -323,6 +323,14 @@ def celeba():
     )
 
 
+def widerface():
+    return collect_download_configs(
+        lambda: datasets.WIDERFace(".", download=True),
+        name=f"WIDERFace",
+        file="widerface",
+    )
+
+
 def make_parametrize_kwargs(download_configs):
     argvalues = []
     ids = []
@@ -357,6 +365,7 @@ def make_parametrize_kwargs(download_configs):
             svhn(),
             usps(),
             celeba(),
+            widerface(),
         )
     )
 )