Browse Source

[FIX] update dataset download URLs

main
troyyyyy 3 months ago
parent
commit
46803a9988
2 changed files with 4 additions and 4 deletions
  1. +3
    -3
      examples/hed/datasets/get_dataset.py
  2. +1
    -1
      examples/hwf/datasets/get_dataset.py

+ 3
- 3
examples/hed/datasets/get_dataset.py View File

@@ -59,8 +59,8 @@ def split_equation(equations_by_len, prop_train, prop_val):
"""
Split the equations in each length to training and validation data according to the proportion
"""
train_equations_by_len = {1: dict(), 0: dict()}
val_equations_by_len = {1: dict(), 0: dict()}
train_equations_by_len = {1: {}, 0: {}}
val_equations_by_len = {1: {}, 0: {}}

for label in range(2):
for equation_len, equations in equations_by_len[label].items():
@@ -80,7 +80,7 @@ def get_dataset(dataset="mnist", train=True):

if not os.path.exists(data_dir):
print("Dataset not exist, downloading it...")
url = "https://drive.google.com/u/0/uc?id=1XoJDjO3cNUdytqVgXUKOBe9dOcUBobom&export=download"
url = "https://drive.google.com/u/0/uc?id=1W2AUn_fnXa4XkgLk4d17K3bEgpae8GMg&export=download"
download_and_unzip(url, os.path.join(CURRENT_DIR, "HED.zip"))
print("Download and extraction complete.")



+ 1
- 1
examples/hwf/datasets/get_dataset.py View File

@@ -31,7 +31,7 @@ def get_dataset(train=True, get_pseudo_label=False):

if not os.path.exists(data_dir):
print("Dataset not exist, downloading it...")
url = "https://drive.google.com/u/0/uc?id=1G07kw-wK-rqbg_85tuB7FNfA49q8lvoy&export=download"
url = "https://drive.google.com/u/0/uc?id=1t52OE2Wdm5GdShX1jD2Wy8phCllk0r8I&export=download"
download_and_unzip(url, os.path.join(CURRENT_DIR, "HWF.zip"))
print("Download and extraction complete.")



Loading…
Cancel
Save