Spaces:
Running
Running
Commit ·
0dadee8
1
Parent(s): 06a9c4f
fix: correct DATASETS dict, genimage->download_genimage_subset
Browse files
scripts/datasets/download_ai.py
CHANGED
|
@@ -326,37 +326,6 @@ def download_genimage_subset():
|
|
| 326 |
add_to_manifest(rows)
|
| 327 |
print(f"Defactify done: {len(rows)} AI images added.")
|
| 328 |
|
| 329 |
-
|
| 330 |
-
|
| 331 |
-
DATASETS = {
|
| 332 |
-
"cifake": download_cifake,
|
| 333 |
-
"faces": download_fake_real_faces,
|
| 334 |
-
"diffusiondb": download_diffusiondb,
|
| 335 |
-
"genimage": download_genimage_subset,
|
| 336 |
-
"all": None,
|
| 337 |
-
}
|
| 338 |
-
|
| 339 |
-
if __name__ == "__main__":
|
| 340 |
-
import argparse
|
| 341 |
-
parser = argparse.ArgumentParser(description="Download AI-generated datasets")
|
| 342 |
-
parser.add_argument(
|
| 343 |
-
"--dataset",
|
| 344 |
-
choices=list(DATASETS.keys()),
|
| 345 |
-
required=True,
|
| 346 |
-
help="Which dataset to download"
|
| 347 |
-
)
|
| 348 |
-
args = parser.parse_args()
|
| 349 |
-
|
| 350 |
-
if args.dataset == "all":
|
| 351 |
-
download_cifake()
|
| 352 |
-
download_fake_real_faces()
|
| 353 |
-
download_diffusiondb()
|
| 354 |
-
download_genimage_subset()
|
| 355 |
-
else:
|
| 356 |
-
DATASETS[args.dataset]()
|
| 357 |
-
|
| 358 |
-
print("\nDone. Check data/manifest.csv")
|
| 359 |
-
|
| 360 |
DATASETS = {
|
| 361 |
"cifake": download_cifake,
|
| 362 |
"faces": download_fake_real_faces,
|
|
|
|
| 326 |
add_to_manifest(rows)
|
| 327 |
print(f"Defactify done: {len(rows)} AI images added.")
|
| 328 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 329 |
DATASETS = {
|
| 330 |
"cifake": download_cifake,
|
| 331 |
"faces": download_fake_real_faces,
|